freedak/timesfm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nan handling

Browse Source
This commit is contained in:
Yichen Zhou
2024-12-20 14:18:24 -08:00
parent 562187c6d1
commit faf1c7681e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/timesfm/timesfm_base.py
+2 -2
View File
@@ -365,8 +365,8 @@ class TimesFmBase:
stats = None stats = None
tmp_inputs = [] tmp_inputs = []
for each_inputs in inputs: for each_input in inputs:
arr = np.array(each_inputs) arr = np.array(each_input)
if not np.isfinite(arr).all(): if not np.isfinite(arr).all():
arr = np.where(np.isfinite(arr), arr, np.nan) arr = np.where(np.isfinite(arr), arr, np.nan)
arr = strip_leading_nans(arr) arr = strip_leading_nans(arr)