fix assert error

drift-labs · May 3, 2022 · 82a47fa · 82a47fa
1 parent a7ce216
commit 82a47fa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/driftpy/clearing_house_user.py b/src/driftpy/clearing_house_user.py
@@ -90,7 +90,7 @@ async def get_user_position(self, market_index: int) -> MarketPosition:
         )
 
     async def get_unrealised_pnl(self, market_index: int=None):
-        assert(market_index is None or int(market_index))
+        assert(market_index is None or int(market_index) >= 0)
         positions_account = await self.get_user_positions_account()
 
         pnl = 0
@@ -124,7 +124,7 @@ async def get_total_position_value(self):
         return value
 
     async def get_position_value(self, market_index: int=None):
-        assert(market_index is None or int(market_index))
+        assert(market_index is None or int(market_index) >= 0)
         positions_account = await self.get_user_positions_account()
         value = 0
         for position in positions_account.positions: