nan handling

This commit is contained in:
Yichen Zhou
2024-12-20 14:18:24 -08:00
parent 562187c6d1
commit faf1c7681e
+2 -2
View File
@@ -365,8 +365,8 @@ class TimesFmBase:
stats = None
tmp_inputs = []
for each_inputs in inputs:
arr = np.array(each_inputs)
for each_input in inputs:
arr = np.array(each_input)
if not np.isfinite(arr).all():
arr = np.where(np.isfinite(arr), arr, np.nan)
arr = strip_leading_nans(arr)