time 0 corrected

deployment/nbeats/predict.py

@@ -96,11 +96,10 @@ def main():
 
     tl = Timeloop()
 
-    time_0 = msg["epoch_start"]
-
     dataset_preprocessor.prepare_csv()
     for metric in predicted_metrics:
         predictions = None
+        global time_0
         time_0 = time_0 + prediction_cycle * 1000
         for i in range(number_of_forward_predictions[metric]):
             prediction_msgs, prediction = predict(
@@ -165,6 +164,7 @@ if __name__ == "__main__":
         }
         for m in msg["all_metrics"]
     }
+    time_0 = msg["epoch_start"]
     prediction_horizon = msg["prediction_horizon"] * 1000
     predicted_metrics = set(msg["metrics"])
     prediction_cycle = msg["prediction_horizon"]

deployment/tft/predict.py

@@ -96,11 +96,10 @@ def main():
 
     tl = Timeloop()
 
-    time_0 = msg["epoch_start"]
-
     dataset_preprocessor.prepare_csv()
     for metric in predicted_metrics:
         predictions = None
+        global time_0
         time_0 = time_0 + prediction_cycle * 1000
         for i in range(number_of_forward_predictions[metric]):
             prediction_msgs, prediction = predict(
@@ -165,6 +164,7 @@ if __name__ == "__main__":
         }
         for m in msg["all_metrics"]
     }
+    time_0 = msg["epoch_start"]
     prediction_horizon = msg["prediction_horizon"] * 1000
     predicted_metrics = set(msg["metrics"])
     prediction_cycle = msg["prediction_horizon"]