From 28f6ed1dbc57b50f01ba2413a10afe5d884cd0d7 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Thu, 16 Jan 2020 10:41:38 +0100 Subject: [PATCH] fix typo --- pts/dataset/multivariate_grouper.py | 1 + pts/model/estimator.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) 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