1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 00:06:58 +08:00

mypy issue

This commit is contained in:
Huoran Li
2022-07-26 16:42:45 +08:00
parent f269274408
commit e4532908db

View File

@@ -59,7 +59,7 @@ def is_invalid(arr: int | float | bool | np.ndarray | dict | list | tuple) -> bo
if isinstance(arr, np.ndarray): if isinstance(arr, np.ndarray):
if np.issubdtype(arr.dtype, np.floating): if np.issubdtype(arr.dtype, np.floating):
return np.isnan(arr).all() 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): 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)):