diff --git a/pts/dataset/artificial.py b/pts/dataset/artificial.py index 0bcaac9..b490872 100644 --- a/pts/dataset/artificial.py +++ b/pts/dataset/artificial.py @@ -62,7 +62,7 @@ def constant_dataset() -> Tuple[DatasetInfo, Dataset, Dataset]: info = DatasetInfo( name="constant_dataset", metadata=metadata, - prediction_length=2, + prediction_length=6, train_statistics=calculate_dataset_statistics(train_ds), test_statistics=calculate_dataset_statistics(test_ds), )