diff --git a/deployment/nbeats/main.py b/deployment/nbeats/main.py index 9b9bea1fa2412cd8efab57c74824ca90456bbedc..4427fce4c7b7bfdfe61af0af79b58b5e9813159c 100644 --- a/deployment/nbeats/main.py +++ b/deployment/nbeats/main.py @@ -114,11 +114,11 @@ def main(): ) # msg1 = Msg() - # msg1.body = '[{"metric": "memory", "level": 3, "publish_rate": 10000}]' + # msg1.body = '[{"metric": "memory", "level": 3, "publish_rate": 30000}]' # msg2 = Msg() # msg2.body = ( # "{" - # + f'"metrics": ["memory"],"timestamp": {int(time.time())}, "epoch_start": {int(time.time()) + 30}, "number_of_forward_predictions": 8,"prediction_horizon": 20' + # + f'"metrics": ["memory"],"timestamp": {int(time.time())}, "epoch_start": {int(time.time()) + 30}, "number_of_forward_predictions": 8,"prediction_horizon": 30' # + "}" # ) diff --git a/deployment/nbeats/predict.py b/deployment/nbeats/predict.py index 5fa7410f9dca5319c2d09d894913cf0e22cdb0ce..2dd0091d5d75ef89739190384fdab82fcd1d85fb 100644 --- a/deployment/nbeats/predict.py +++ b/deployment/nbeats/predict.py @@ -126,7 +126,7 @@ def main(): prediction_msgs, prediction = predict( metric, prediction_length, - extra_data=predictions, + extra_data=None, m=i + 1, prediction_hor=prediction_horizon, timestamp=time_0 + (i + 1) * (prediction_horizon // 1000), @@ -136,10 +136,7 @@ def main(): print( f"time_0 difference seconds {start_time + (i + 1) * prediction_horizon // 1000 - int(time.time())}" ) - # if predictions is not None: - # predictions = pd.concat( - # [predictions, prediction], ignore_index=True - # ) + else: predictions = prediction diff --git a/deployment/tft/predict.py b/deployment/tft/predict.py index c56c10c8eb6bb373523d1fe2ffc668eee3e07aaa..dc057f4219a9716bd7da1cbf11ee968111531558 100644 --- a/deployment/tft/predict.py +++ b/deployment/tft/predict.py @@ -126,7 +126,7 @@ def main(): prediction_msgs, prediction = predict( metric, prediction_length, - extra_data=predictions, + extra_data=None, m=i + 1, prediction_hor=prediction_horizon, timestamp=time_0 + (i + 1) * (prediction_horizon // 1000),