Bump PMAT version (#35)

* Bump PMAT version
gnosis · Apr 2, 2024 · cf3c2a1 · cf3c2a1
1 parent c02be15
commit cf3c2a1
Show file tree

Hide file tree

Showing 5 changed files with 99 additions and 95 deletions.
diff --git a/main.py b/main.py
@@ -2,8 +2,10 @@
 
 import typer
 from prediction_market_agent_tooling.markets.agent_market import SortBy
-from prediction_market_agent_tooling.markets.markets import MARKET_TYPE_MAP, MarketType
-from prediction_market_agent_tooling.tools.utils import check_not_none
+from prediction_market_agent_tooling.markets.markets import (
+    MarketType,
+    get_binary_markets,
+)
 
 import prediction_market_agent as pma
 from prediction_market_agent.agents.all_agents import AgentType, get_agent
@@ -18,8 +20,11 @@ def main(
     Picks one market and answers it, optionally placing a bet.
     """
     # Pick a market
-    cls = check_not_none(MARKET_TYPE_MAP.get(market_type))
-    market = cls.get_binary_markets(limit=1, sort_by=SortBy.NEWEST)[0]
+    market = get_binary_markets(
+        limit=1,
+        sort_by=SortBy.NEWEST,
+        market_type=market_type,
+    )[0]
 
     # Create the agent and run it
     agent = get_agent(agent_type)