some fix

qianyun210603 · Dec 12, 2023 · 0c7e2e7 · 0c7e2e7
1 parent 9879fff
commit 0c7e2e7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/qlib/backtest/exchange.py b/qlib/backtest/exchange.py
@@ -325,7 +325,7 @@ def get_quote_from_qlib(self, subscribe_fields_alias: Dict[str, str] = None) ->
     LT_FLT = "float"  # float:  the trading limitation is based on `abs($change) < limit_threshold`
     LT_NONE = "none"  # none:  there is no trading limitation
 
-    def get_instrument_info(self, symbol: str) -> Optional[Dict]:
+    def get_instrument_info(self, symbol: str) -> Any:
         return self.instrument_info.get(symbol, None)
 
     def _get_limit_type(self) -> str:

diff --git a/qlib/backtest/position.py b/qlib/backtest/position.py
@@ -359,6 +359,7 @@ def _buy_stock(self, stock_id: str, trade_val: float, cost: float, trade_price:
         else:
             # exist, add amount
             self.position[stock_id]["amount"] += trade_amount
+            self.position[stock_id]["cost_basis"] += trade_val + cost
 
         self.position["cash"] -= trade_val + cost