fix mypy

qianyun210603 · Oct 2, 2023 · 79c2d5a · 79c2d5a
1 parent e3dee30
commit 79c2d5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qlib/backtest/exchange.py b/qlib/backtest/exchange.py
@@ -482,7 +482,7 @@ def check_stock_suspended(
                 return cast(bool, idx_data.isna().all() or np.isclose(idx_data.data, 0).all())
             else:
                 # it is single value, make sure is not None
-                return np.isnan(volume) or np.isclose(volume, 0)
+                return cast(bool, np.isnan(volume) or np.isclose(volume, 0))
         else:
             # if the stock is not in the stock list, then it is not tradable and regarded as suspended
             return True