Adding is_correct to model_dump

gnosis · Jul 1, 2024 · 19e5790 · 19e5790
1 parent f9f5d8a
commit 19e5790
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/prediction_market_agent_tooling/markets/data_models.py b/prediction_market_agent_tooling/markets/data_models.py
@@ -2,7 +2,7 @@
 from enum import Enum
 from typing import TypeAlias
 
-from pydantic import BaseModel
+from pydantic import BaseModel, computed_field
 
 from prediction_market_agent_tooling.gtypes import OutcomeStr
 
@@ -41,6 +41,7 @@ class ResolvedBet(Bet):
     resolved_time: datetime
     profit: ProfitAmount
 
+    @computed_field
     @property
     def is_correct(self) -> bool:
         return self.outcome == self.market_outcome