From 754c91a116f196b20544a648e4ba895a2cdedb90 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Santeri=20Hyv=C3=A4rinen?= Date: Tue, 13 May 2025 10:06:25 +0000 Subject: [PATCH] Fix series conversion to comply current darts.dataset interface --- data/small_context.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data/small_context.py b/data/small_context.py index 853ce7d..14551b6 100644 --- a/data/small_context.py +++ b/data/small_context.py @@ -58,7 +58,7 @@ def get_dataset(dsname): darts_ds = getattr(darts.datasets,dsname)().load() if dsname=='GasRateCO2Dataset': darts_ds = darts_ds[darts_ds.columns[1]] - series = darts_ds.pd_series() + series = darts_ds.to_series() if dsname == 'SunspotsDataset': series = series.iloc[::4] @@ -111,4 +111,4 @@ def get_memorization_datasets(n=-1,testfrac=0.15, predict_steps=30): datas.append((train,test)) if i+1==n: break - return dict(zip(datasets,datas)) \ No newline at end of file + return dict(zip(datasets,datas))