update for ExtendedCurveHistory

drift-labs · Jan 30, 2022 · cac681f · cac681f
1 parent 471ade2
commit cac681f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/driftpy/admin.py b/src/driftpy/admin.py
@@ -121,7 +121,9 @@ async def initialize(
                     program.account["DepositHistory"].create_instruction(
                         deposit_history
                     ),
-                    program.account["CurveHistory"].create_instruction(curve_history),
+                    program.account["ExtendedCurveHistory"].create_instruction(
+                        curve_history
+                    ),
                 ),
                 signers=[
                     deposit_history,

diff --git a/src/driftpy/clearing_house.py b/src/driftpy/clearing_house.py
@@ -20,7 +20,7 @@
     TradeHistoryAccount,
     LiquidationHistoryAccount,
     DepositHistoryAccount,
-    CurveHistoryAccount,
+    ExtendedCurveHistoryAccount,
     User,
     UserPositions,
 )
@@ -119,9 +119,9 @@ async def get_deposit_history_account(self) -> DepositHistoryAccount:
         )
         return cast(DepositHistoryAccount, res)
 
-    async def get_curve_history_account(self) -> CurveHistoryAccount:
+    async def get_curve_history_account(self) -> ExtendedCurveHistoryAccount:
         res = await self.program.account["CurveHistory"].fetch(self.pdas.curve_history)
-        return cast(CurveHistoryAccount, res)
+        return cast(ExtendedCurveHistoryAccount, res)
 
     @staticmethod
     def _get_state_pubkey(program: Program) -> PublicKey: