From cb441c03850bd4744e334d67cbe6446ddf26a42f Mon Sep 17 00:00:00 2001 From: JXZhou Date: Fri, 17 Jan 2025 15:34:43 +0800 Subject: [PATCH] fix mypy errors in sotopia/database --- sotopia/database/logs.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/sotopia/database/logs.py b/sotopia/database/logs.py index 3eec070d..da66c7fc 100644 --- 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")