diff --git a/pts/dataset/multivariate_grouper.py b/pts/dataset/multivariate_grouper.py index 25b88e4..9b2f369 100644 --- a/pts/dataset/multivariate_grouper.py +++ b/pts/dataset/multivariate_grouper.py @@ -204,3 +204,4 @@ class MultivariateGrouper: freq=data[FieldName.START].freq, ), ) + \ No newline at end of file diff --git a/pts/model/estimator.py b/pts/model/estimator.py index ee8b4b9..8eba3a6 100644 --- a/pts/model/estimator.py +++ b/pts/model/estimator.py @@ -108,7 +108,7 @@ class PTSEstimator(Estimator): ) training_data_loader = DataLoader( - traning_iter_dataset, + training_iter_dataset, batch_size=self.trainer.batch_size, num_workers=self.train.num_workers, pip_memory=self.trainer.pip_memory