diff --git a/pts/dataset/artificial.py b/pts/dataset/artificial.py index de36bd9..fdc3d4f 100644 --- a/pts/dataset/artificial.py +++ b/pts/dataset/artificial.py @@ -72,7 +72,7 @@ class ArtificialDataset: return TrainDatasets( metadata=self.metadata, train=ListDataset(self.train, self.freq), - test=ListDataset(self.test, self.freq, shuffle=False), + test=ListDataset(self.test, self.freq), ) @@ -697,7 +697,7 @@ class RecipeDataset(ArtificialDataset): return TrainDatasets( metadata=metadata, train=ListDataset(train_data, metadata.freq), - test=ListDataset(test_data, metadata.freq, is_train=False), + test=ListDataset(test_data, metadata.freq), ) diff --git a/pts/dataset/multivariate_grouper.py b/pts/dataset/multivariate_grouper.py index d6aaaa2..765ba8e 100644 --- a/pts/dataset/multivariate_grouper.py +++ b/pts/dataset/multivariate_grouper.py @@ -144,7 +144,7 @@ class MultivariateGrouper: all_entries.append(grouped_data) return ListDataset( - all_entries, freq=self.frequency, one_dim_target=False, is_train=False + all_entries, freq=self.frequency, one_dim_target=False ) def _align_data_entry(self, data: DataEntry) -> np.array: