diff --git a/pytorch_forecasting/data/timeseries.py b/pytorch_forecasting/data/timeseries.py index 0eb7d548..2c9a9643 100644 --- 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