fix: One line X_splits unpacking

functime-org · Dec 2, 2023 · 88c4d63 · 88c4d63
1 parent 6e45b6f
commit 88c4d63
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/functime/forecasting/_evaluate.py b/functime/forecasting/_evaluate.py
@@ -77,7 +77,10 @@ def evaluate_windows(
     results = []
     for i in range(n_splits):
         y_train, y_test = y_splits[i]
-        X_train, X_test = X_splits[i] if X_splits is not None else None, None
+        X_train = None
+        X_test = None
+        if X_splits is not None:
+            X_train, X_test = X_splits[i]
         result = evaluate_window(
             y_train=y_train,
             y_test=y_test,