From 289f2ba3e022eb3c8000618978e5428a5fa70a34 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Wed, 22 Jun 2022 12:14:35 +0200 Subject: [PATCH] fix api change --- tft/estimator.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tft/estimator.py b/tft/estimator.py index c4ac35d..9086c38 100644 --- a/tft/estimator.py +++ b/tft/estimator.py @@ -268,7 +268,6 @@ class TFTEstimator(PyTorchLightningEstimator): input_names=PREDICTION_INPUT_NAMES, prediction_net=module.model, batch_size=self.batch_size, - freq=self.freq, prediction_length=self.prediction_length, device=torch.device("cuda" if torch.cuda.is_available() else "cpu"), )