fix mypy errors in sotopia/database

sotopia-lab · Jan 17, 2025 · cb441c0 · cb441c0
1 parent 02ea368
commit cb441c0
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/sotopia/database/logs.py b/sotopia/database/logs.py
@@ -28,9 +28,7 @@ class BaseEpisodeLog(BaseModel):
     models: list[str] | None = Field(index=True, default=[])
     messages: list[list[tuple[str, str, str]]]  # Messages arranged by turn
     reasoning: str = Field(default="")
-    rewards: list[
-        tuple[float, dict[str, float]] | float | dict[str, dict]
-    ]  # Rewards arranged by turn
+    rewards: list[tuple[float, dict[str, float]] | float]  # Rewards arranged by turn
     rewards_prompt: str
 
     @model_validator(mode="after")