Update timeseries.py

sktime · Jan 4, 2025 · b22b128 · b22b128
1 parent 0637c42
commit b22b128
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pytorch_forecasting/data/timeseries.py b/pytorch_forecasting/data/timeseries.py
@@ -1401,9 +1401,9 @@ def _to_tensor(cols, long=True):
             """
             if not isinstance(cols, list) and cols not in data.columns:
                 return None
-            if isinstance(cols, list):
-                if len(cols) == 0:
-                    return []
+            if isinstance(cols, list) and len(cols) == 0:
+                dtypekind = "f"
+            elif isinstance(cols, list):  # and len(cols) > 0
                 dtypekind = data.dtypes[cols[0]].kind
             else:
                 dtypekind = data.dtypes[cols].kind