Edit rtu pipeline

src/main.py

@@ -2,12 +2,12 @@ import json
 from rtu.RTUAnomalizer import RTUAnomalizer
 from rtu.RTUPipeline import RTUPipeline
 from vav.VAVPipeline import VAVPipeline
+from vav.VAVAnomalizer import VAVAnomalizer
 import paho.mqtt.client as mqtt
 
 
 def main():
     rtu_data_pipeline = RTUPipeline(scaler_path="src/rtu/models/scaler_1.pkl")
-    print(rtu_data_pipeline.scaler)
     rtu_anomalizer = RTUAnomalizer(
         prediction_model_path="src/rtu/models/lstm_4rtu_smooth_02.keras",
         clustering_model_paths=[
@@ -20,8 +20,9 @@ def main():
         num_outputs=rtu_data_pipeline.num_outputs,
     )
 
-    # vav_pipeline = VAVPipeline(rtu_id=1)
+    vav_pipeline = VAVPipeline(rtu_id=1, scaler_path="src/vav/models/scaler_vav_1.pkl")
 
+    vav_anomalizer = VAVAnomalizer(prediction_model_path="src/vav/models/lstm__vav_01")
     # print(vav_pipeline.input_col_names)
 
     # print(len(vav_pipeline.output_col_names))

src/rtu/RTUPipeline.py

@@ -9,39 +9,25 @@ import numpy as np
 class RTUPipeline:
     scaler = None
 
-    def __init__(self, scaler_path=None):
+    def __init__(self, rtus=[1, 2], scaler_path=None):
+
+        outputs = [
+            "sa_temp",
+            "oadmpr_pct",
+            "ra_temp",
+            "oa_temp",
+            "ma_temp",
+            "sf_vfd_spd_fbk_tn",
+            "rf_vfd_spd_fbk_tn",
+        ]
         self.output_col_names = [
             "hp_hws_temp",
-            "rtu_003_sa_temp",
-            "rtu_003_oadmpr_pct",
-            "rtu_003_ra_temp",
-            "rtu_003_oa_temp",
-            "rtu_003_ma_temp",
-            "rtu_003_sf_vfd_spd_fbk_tn",
-            "rtu_003_rf_vfd_spd_fbk_tn",
-            "rtu_004_sa_temp",
-            "rtu_004_oadmpr_pct",
-            "rtu_004_ra_temp",
-            "rtu_004_oa_temp",
-            "rtu_004_ma_temp",
-            "rtu_004_sf_vfd_spd_fbk_tn",
-            "rtu_004_rf_vfd_spd_fbk_tn",
-            "rtu_001_sa_temp",
-            "rtu_001_oadmpr_pct",
-            "rtu_001_ra_temp",
-            "rtu_001_oa_temp",
-            "rtu_001_ma_temp",
-            "rtu_001_sf_vfd_spd_fbk_tn",
-            "rtu_001_rf_vfd_spd_fbk_tn",
-            "rtu_002_sa_temp",
-            "rtu_002_oadmpr_pct",
-            "rtu_002_ra_temp",
-            "rtu_002_oa_temp",
-            "rtu_002_ma_temp",
-            "rtu_002_sf_vfd_spd_fbk_tn",
-            "rtu_002_rf_vfd_spd_fbk_tn",
         ]
 
+        for rtu in rtus:
+            for output in outputs:
+                self.output_col_names.append(f"rtu_00{rtu}_{output}")
+
         self.input_col_names = [
             "air_temp_set_1",
             "air_temp_set_2",