minor bug fix

socialfoundations · Jun 27, 2024 · a7d7bad · a7d7bad
1 parent 9708a28
commit a7d7bad
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/folktexts/cli/launch_experiments.py b/folktexts/cli/launch_experiments.py
@@ -78,7 +78,7 @@
 # Function that defines common settings among all LLM-as-clf experiments
 def make_llm_as_clf_experiment(
     model_name: str,
-    task_name: str,
+    task: str,
     results_root_dir: str,
     **kwargs,
 ) -> Experiment:
@@ -114,7 +114,7 @@ def make_llm_as_clf_experiment(
         executable_path=EXECUTABLE_PATH.as_posix(),
         kwargs=dict(
             model=model_path,
-            task_name=task_name,
+            task=task,
             **experiment_kwargs,
         ),
         **job_kwargs,
@@ -138,7 +138,7 @@ def get_llm_results_folder(exp: Experiment) -> str:
     """
     return (
         f"model-{Path(exp.model).name}."
-        f"dataset-{exp.task_name}"
+        f"dataset-{exp.task}"
     )
 
 
@@ -211,7 +211,7 @@ def main():
         all_experiments = [
             make_llm_as_clf_experiment(
                 model_name=model,
-                task_name=task,
+                task=task,
                 results_root_dir=args.results_root_dir,
                 **extra_kwargs,
             )