fixamount

microsoft · Oct 13, 2023 · 7d32321 · 7d32321
1 parent ecbeeaf
commit 7d32321
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qlib/backtest/position.py b/qlib/backtest/position.py
@@ -354,7 +354,7 @@ def _sell_stock(self, stock_id: str, trade_val: float, cost: float, trade_price:
         if stock_id not in self.position:
             raise KeyError("{} not in current position".format(stock_id))
         else:
-            if np.isclose(self.position[stock_id]["amount"], trade_amount):
+            if np.isclose(self.position[stock_id]["amount"], trade_amount) or self.position[stock_id]["amount"] <= trade_amount:
                 # Selling all the stocks
                 # we use np.isclose instead of abs(<the final amount>) <= 1e-5  because `np.isclose` consider both
                 # relative amount and absolute amount