From 863c6759811204adf2ea0f9d04757655a780f1ad Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Mon, 13 Jan 2020 12:04:11 +0100 Subject: [PATCH] rename TODO remove and refactor arg. names --- pts/model/tempflow/tempflow_estimator.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/pts/model/tempflow/tempflow_estimator.py b/pts/model/tempflow/tempflow_estimator.py index 70b03a2..62f6f74 100644 --- a/pts/model/tempflow/tempflow_estimator.py +++ b/pts/model/tempflow/tempflow_estimator.py @@ -153,6 +153,13 @@ class TempFlowEstimator(PTSEstimator): ], pick_incomplete=self.pick_incomplete, ), + RenameFields( + { + f"past_{FieldName.TARGET}": f"past_{FieldName.TARGET}_cdf", + f"future_{FieldName.TARGET}": f"future_{FieldName.TARGET}_cdf", + } + ) + ] )