Merge branch 'staging' into pre-staging

macrocosm-os · Sep 4, 2024 · 0f6a538 · 0f6a538
2 parents 36cd277 + 1538387
commit 0f6a538
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 341 deletions.
diff --git a/prompting/mock.py b/prompting/mock.py
diff --git a/prompting/tasks/task_registry.py b/prompting/tasks/task_registry.py
@@ -42,9 +42,7 @@ class TaskRegistry(BaseModel):
             datasets=[DDGDataset],
             reward_model=InferenceRewardConfig,
         ),
-        TaskConfig(
-            task=MultiChoiceTask, probability=0.2, datasets=[WikiDataset], reward_model=MultiChoiceRewardConfig
-        ),
+        TaskConfig(task=MultiChoiceTask, probability=0.2, datasets=[WikiDataset], reward_model=MultiChoiceRewardConfig),
     ]
 
     @classmethod