Fixed calculate_remove_liquidity so it always returns nft_ids

galacticcouncil · Sep 23, 2024 · 2a9b7b4 · 2a9b7b4
1 parent e807eb5
commit 2a9b7b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hydradx/model/amm/omnipool_amm.py b/hydradx/model/amm/omnipool_amm.py
@@ -906,7 +906,7 @@ def calculate_remove_liquidity(self, agent: Agent, quantity: float = None, tkn_r
         if quantity is not None:
             if nft_id is None:  # remove specified quantity of shares from holdings
                 k = (self.unique_id, tkn_remove)
-                return self._calculate_remove_one_position(
+                delta_qa, delta_r, delta_q, delta_s, delta_b, delta_l = self._calculate_remove_one_position(
                     quantity=quantity, tkn_remove=tkn_remove, share_price=agent.share_prices[k]
                 )
             else:  # remove specified quantity of shares from specified position