fixed assert error (#1354)

ourownstory · Jun 14, 2023 · 1930e44 · 1930e44
1 parent 5b7ab72
commit 1930e44
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/neuralprophet/time_net.py b/neuralprophet/time_net.py
@@ -645,6 +645,7 @@ def forward(self, inputs: Dict, meta: Dict = None, compute_components: bool = Fa
             prediction = self._forward(corrected_inputs, meta, non_stationary_only=False)
         else:
             prediction = self._forward(inputs, meta)
+            corrected_inputs = inputs.copy()
 
         # check for crossing quantiles and correct them here
         if "predict_mode" in inputs.keys() and inputs["predict_mode"]:
@@ -654,8 +655,6 @@ def forward(self, inputs: Dict, meta: Dict = None, compute_components: bool = Fa
         prediction_with_quantiles = self._compute_quantile_forecasts_from_diffs(prediction, predict_mode)
 
         # compute components
-        corrected_inputs = inputs if corrected_inputs is None else corrected_inputs
-
         if compute_components:
             components = self.compute_components(corrected_inputs, meta)
             return prediction_with_quantiles, components