Removed 'arbitrary_types_allowed=True' from Pydantic configs (#11)

Future-House · Sep 3, 2024 · d8981ab · d8981ab
1 parent 48c5ef6
commit d8981ab
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/packages/gsm8k/src/aviary/gsm8k/env.py b/packages/gsm8k/src/aviary/gsm8k/env.py
@@ -18,7 +18,7 @@
 
 
 class CalculatorEnvConfig(BaseModel):
-    model_config = ConfigDict(extra="forbid", arbitrary_types_allowed=True)
+    model_config = ConfigDict(extra="forbid")
 
     correct_reward: float = 1.0
     incorrect_reward: float = 0.0

diff --git a/packages/hotpotqa/src/aviary/hotpotqa/env.py b/packages/hotpotqa/src/aviary/hotpotqa/env.py
@@ -467,7 +467,7 @@ class HotPotQAEnvConfig(BaseModel):
     including the path to the data repository and the specific path to the prompt file.
     """
 
-    model_config = ConfigDict(extra="forbid", arbitrary_types_allowed=True)
+    model_config = ConfigDict(extra="forbid")
 
     shuffle_data: bool = Field(
         default=False, description="Set True to shuffle the dataset after loading."

diff --git a/src/aviary/env.py b/src/aviary/env.py
@@ -326,7 +326,7 @@ def iter_batches(
 class TaskConfig(BaseModel):
     """Convenience for making a config file entry for a TaskDataset."""
 
-    model_config = ConfigDict(extra="forbid", arbitrary_types_allowed=True)
+    model_config = ConfigDict(extra="forbid")
 
     name: str
     task_kwargs: dict[str, BaseModel | JsonValue] = Field(