drift_client.py: fix get_oracle_price_data_and_slot typo

drift-labs · Dec 2, 2023 · 123fb5f · 123fb5f
1 parent c718ec4
commit 123fb5f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/driftpy/drift_client.py b/src/driftpy/drift_client.py
@@ -188,7 +188,7 @@ def get_spot_market_account(self, market_index: int) -> Optional[SpotMarketAccou
         return getattr(spot_market_and_slot, "data", None)
 
     def get_oracle_price_data(self, oracle: Pubkey) -> Optional[OraclePriceData]:
-        oracle_price_data_and_slot = self.account_subscriber.get_oracle_data_and_slot(
+        oracle_price_data_and_slot = self.account_subscriber.get_oracle_price_data_and_slot(
             oracle
         )
         return getattr(oracle_price_data_and_slot, "data", None)

diff --git a/src/driftpy/math/positions.py b/src/driftpy/math/positions.py
@@ -1,6 +1,7 @@
 from driftpy.math.spot_market import *
 from driftpy.types import OraclePriceData
-
+from driftpy.constants.numeric_constants import *
+from driftpy.math.amm import calculate_amm_reserves_after_swap, get_swap_direction
 
 def get_worst_case_token_amounts(
     position: SpotPosition,