one more add

gnosis · Feb 19, 2024 · 701f03e · 701f03e
1 parent 23e5483
commit 701f03e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/prediction_market_agent_tooling/benchmark/utils.py b/prediction_market_agent_tooling/benchmark/utils.py
@@ -261,11 +261,14 @@ def get_polymarket_markets(
 def get_markets(
     number: int,
     source: MarketSource,
+    excluded_questions: set[str] | None = None,
 ) -> t.List[Market]:
     if source == MarketSource.MANIFOLD:
-        return get_manifold_markets(limit=number)
+        return get_manifold_markets_paged(number=number, excluded_questions=excluded_questions)
     elif source == MarketSource.POLYMARKET:
-        return get_polymarket_markets(limit=number)
+        return get_polymarket_markets(
+            limit=number, excluded_questions=excluded_questions
+        )
     else:
         raise ValueError(f"Unknown market source: {source}")