From e4532908db5ef38e45b8e1f6027f5e3e6ec32b9a Mon Sep 17 00:00:00 2001 From: Huoran Li Date: Tue, 26 Jul 2022 16:42:45 +0800 Subject: [PATCH] mypy issue --- qlib/rl/utils/finite_env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)):