Fixing run.py to address invest_all behavior

galacticcouncil · Jun 27, 2023 · e3ab50f · e3ab50f
1 parent 6d048af
commit e3ab50f
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/hydradx/model/run.py b/hydradx/model/run.py
@@ -1,6 +1,7 @@
 import time
 
 from .amm.global_state import GlobalState
+from copy import deepcopy
 
 
 def run(initial_state: GlobalState, time_steps: int, silent: bool = False) -> list:
@@ -21,16 +22,11 @@ def run(initial_state: GlobalState, time_steps: int, silent: bool = False) -> li
         # market evolutions
         new_global_state.evolve()
 
-        # agent actions
-        # agents = new_global_state.agents
-        # for agent_id, agent in agents.items():
-        #     if agent.trade_strategy:
-        #         new_global_state = agent.trade_strategy.execute(new_global_state, agent.unique_id)
-
-        for agent_id in new_global_state.agents:
-            # works as long as no trade_strategy adds or removes agents
-            if new_global_state.agents[agent_id].trade_strategy:
-                new_global_state = new_global_state.agents[agent_id].trade_strategy.execute(new_global_state, agent_id)
+        agent_ids = deepcopy(new_global_state.agents.keys())
+        for agent_id in agent_ids:
+            agent = new_global_state.agents[agent_id]
+            if agent.trade_strategy:
+                new_global_state = agent.trade_strategy.execute(new_global_state, agent.unique_id)
 
         events.append(new_global_state.archive())