liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 16:26:55 +08:00
Code Activity

Mypy issue

Browse Source
This commit is contained in:
Huoran Li
2022-07-26 16:27:27 +08:00
parent ccc3f96ed3
commit f269274408
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
qlib/rl/utils/finite_env.py
View File

@@ -56,11 +56,10 @@ def fill_invalid(obj: int | float | bool | np.ndarray | dict | list | tuple) ->
def is_invalid(arr: int | float | bool | np.ndarray | dict | list | tuple) -> bool: def is_invalid(arr: int | float | bool | np.ndarray | dict | list | tuple) -> bool:
if hasattr(arr, "dtype"): if isinstance(arr, np.ndarray):
dtype = getattr(arr, "dtype") if np.issubdtype(arr.dtype, np.floating):
if np.issubdtype(dtype, np.floating):
return np.isnan(arr).all() return np.isnan(arr).all()
return (np.iinfo(dtype).max == arr).all() return (np.iinfo(arr.dtype).max == arr).all()
if isinstance(arr, dict): if isinstance(arr, dict):
return all(is_invalid(o) for o in arr.values()) return all(is_invalid(o) for o in arr.values())
if isinstance(arr, (list, tuple)): if isinstance(arr, (list, tuple)):