From faf1c7681ec3a95bcfb178f07ba9a4f839e9b77c Mon Sep 17 00:00:00 2001 From: Yichen Zhou Date: Fri, 20 Dec 2024 14:18:24 -0800 Subject: [PATCH] nan handling --- src/timesfm/timesfm_base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/timesfm/timesfm_base.py b/src/timesfm/timesfm_base.py index 260eeba..19b92da 100644 --- a/src/timesfm/timesfm_base.py +++ b/src/timesfm/timesfm_base.py @@ -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)