minor fix to benchmark launcher script

socialfoundations · Jun 25, 2024 · b9c69af · b9c69af
1 parent 574eb71
commit b9c69af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/folktexts/benchmark.py b/folktexts/benchmark.py
@@ -384,14 +384,14 @@ def make_acs_benchmark(
         # Fetch ACS task and dataset
         acs_task = ACSTaskMetadata.get_task(task_name)
         acs_dataset = ACSDataset.make_from_task(
-            task_obj=acs_task,
+            task=acs_task,
             cache_dir=data_dir,
             **acs_dataset_configs)
 
         return cls.make_benchmark(
             model=model,
             tokenizer=tokenizer,
-            task_obj=acs_task,
+            task=acs_task,
             dataset=acs_dataset,
             config=config,
         )