diff --git a/qlib/rl/utils/finite_env.py b/qlib/rl/utils/finite_env.py index d3dbfa1f7..806cb5111 100644 --- a/qlib/rl/utils/finite_env.py +++ b/qlib/rl/utils/finite_env.py @@ -59,7 +59,7 @@ def is_invalid(arr: int | float | bool | np.ndarray | dict | list | tuple) -> bo if isinstance(arr, np.ndarray): if np.issubdtype(arr.dtype, np.floating): return np.isnan(arr).all() - return (np.iinfo(arr.dtype).max == arr).all() + return cast(np.ndarray, np.iinfo(arr.dtype).max == arr).all() if isinstance(arr, dict): return all(is_invalid(o) for o in arr.values()) if isinstance(arr, (list, tuple)):