diff --git a/pts/model/tempflow/tempflow_estimator.py b/pts/model/tempflow/tempflow_estimator.py index 62f6f74..59d921b 100644 --- a/pts/model/tempflow/tempflow_estimator.py +++ b/pts/model/tempflow/tempflow_estimator.py @@ -222,5 +222,5 @@ class TempFlowEstimator(PTSEstimator): freq=self.freq, prediction_length=self.prediction_length, device=device, - output_transform=self.output_transform, + output_transform=None, )