diff --git a/pts/model/tempflow/tempflow_estimator.py b/pts/model/tempflow/tempflow_estimator.py index 2dbc848..c0ac17f 100644 --- a/pts/model/tempflow/tempflow_estimator.py +++ b/pts/model/tempflow/tempflow_estimator.py @@ -179,7 +179,7 @@ class TempFlowEstimator(PTSEstimator): conditioning_length=self.conditioning_length, ).to(device) - def create_predictor( + def create_predictor( self, transformation: Transformation, trained_network: TempFlowTrainingNetwork,