Fixing Seer tests

gnosis · Dec 26, 2024 · 4eb7cc5 · 4eb7cc5
1 parent 7f86266
commit 4eb7cc5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/prediction_market_agent_tooling/markets/seer/data_models.py b/prediction_market_agent_tooling/markets/seer/data_models.py
@@ -3,13 +3,17 @@
 from prediction_market_agent_tooling.gtypes import HexBytes
 
 
+class SeerParentMarket(BaseModel):
+    id: HexBytes
+
+
 class SeerMarket(BaseModel):
     model_config = ConfigDict(populate_by_name=True)
 
     id: HexBytes
     title: str = Field(alias="marketName")
     outcomes: list[str]
-    parent_market: HexBytes | None = Field(alias="parentMarket")
+    parent_market: SeerParentMarket | None = Field(alias="parentMarket")
     wrapped_tokens: list[HexBytes] = Field(alias="wrappedTokens")
 
 

diff --git a/prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py b/prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py
@@ -47,7 +47,7 @@ def _get_fields_for_markets(self, markets_field: FieldPath) -> list[FieldPath]:
             markets_field.creator,
             markets_field.marketName,
             markets_field.outcomes,
-            markets_field.parentMarket,
+            markets_field.parentMarket.id,
             markets_field.finalizeTs,
             markets_field.wrappedTokens,
         ]

diff --git a/tests_integration/markets/seer/test_seer_subgraph_handler.py b/tests_integration/markets/seer/test_seer_subgraph_handler.py
@@ -17,7 +17,6 @@ def handler() -> t.Generator[SeerSubgraphHandler, None, None]:
     yield SeerSubgraphHandler()
 
 
-@pytest.skip(reason="Seer subgraph ")
 def test_get_all_seer_markets(handler: SeerSubgraphHandler) -> None:
     markets = handler.get_bicategorical_markets()
     assert len(markets) > 1