Commit 1a183d03 authored by Michael Pieler's avatar Michael Pieler
Browse files

fix: copy paste mistake

parent 1964ccd3
...@@ -2,6 +2,8 @@ import argparse ...@@ -2,6 +2,8 @@ import argparse
import numpy as np import numpy as np
import json import json
import os import os
import random
from lm_eval import tasks
from lm_eval.utils import join_iters from lm_eval.utils import join_iters
from lm_eval.tasks import include_path from lm_eval.tasks import include_path
from lm_eval.logger import eval_logger from lm_eval.logger import eval_logger
...@@ -39,8 +41,8 @@ def main(): ...@@ -39,8 +41,8 @@ def main():
else: else:
task_names = args.tasks.split(",") task_names = args.tasks.split(",")
task_dict = tasks.get_task_dict(task_names) task_dict = tasks.get_task_dict(task_names)
os.makedirs(args.output_base_path, exist_ok=True) os.makedirs(args.output_base_path, exist_ok=True)
for task_name, task in task_dict.items(): for task_name, task in task_dict.items():
rnd = random.Random() rnd = random.Random()
rnd.seed(args.seed) rnd.seed(args.seed)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment