corrected back_and_forth trade strategy

galacticcouncil · Jul 12, 2023 · 0db222f · 0db222f
1 parent 2d70410
commit 0db222f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/hydradx/model/amm/trade_strategies.py b/hydradx/model/amm/trade_strategies.py
@@ -129,8 +129,9 @@ def back_and_forth(
     def strategy(state: GlobalState, agent_id: str):
         omnipool: OmnipoolState = state.pools[pool_id]
         agent: Agent = state.agents[agent_id]
-        for i in range(len(agent.asset_list)):
-            asset = omnipool.asset_list[i]
+        assets = list(set(agent.asset_list) & set(omnipool.asset_list))
+        for asset in assets:
+            # asset = agent.asset_list[i]
             dr = percentage / 2 * omnipool.liquidity[asset]
             lrna_init = state.agents[agent_id].holdings['LRNA']
             oamm.execute_swap(omnipool, agent, tkn_sell=asset, tkn_buy='LRNA', sell_quantity=dr, modify_imbalance=False)