Minor bug fixes for streamlit app

mpkrass7 · Jul 23, 2024 · a986a0f · a986a0f
1 parent 7c8ef54
commit a986a0f
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 8 deletions.
diff --git a/YoutubeForecastMaker/conf/base/parameters.yml b/YoutubeForecastMaker/conf/base/parameters.yml
@@ -85,17 +85,17 @@ deploy_forecast:
         runLeakageRemovedFeatureList: true
       trigger:
         type: accuracy_decline
-        schedule: 
-          minute: 
-            - 0
+        schedule:
+          minute:
+          - 0
           hour:
-            - 0
+          - 0
           dayOfMonth:
-            - "*"
+          - '*'
           month:
-            - "*"
+          - '*'
           dayOfWeek:
-            - "*"
+          - '*'
         statusDeclinesToWarning: true
         statusDeclinesToFailing: false
         statusStillInDecline: false

diff --git a/YoutubeForecastMaker/src/YoutubeForecastMaker/pipelines/deploy_forecast/nodes.py b/YoutubeForecastMaker/src/YoutubeForecastMaker/pipelines/deploy_forecast/nodes.py
@@ -130,6 +130,7 @@ def ensure_deployment_settings(
 
     if association_id is None:
         request_body["associationId"] = {
+            "columnNames": ["association_id"],
             "requiredInPredictionRequests": False,
             "autoGenerateId": True,
         }

diff --git a/YoutubeForecastMaker/src/YoutubeForecastMaker/pipelines/deploy_streamlit_app/pipeline.py b/YoutubeForecastMaker/src/YoutubeForecastMaker/pipelines/deploy_streamlit_app/pipeline.py
@@ -94,7 +94,7 @@ def create_pipeline(**kwargs) -> Pipeline:
                 "datetime_partition_column": "params:project.datetime_partitioning_config.datetime_partition_column",
                 "multiseries_id_column": "params:project.datetime_partitioning_config.multiseries_id_columns",
                 "prediction_interval": "params:deployment.prediction_interval",
-                "prediction_data": "prediction_data_id"
+                "scoring_data": "prediction_data_id"
             },
             outputs="app_parameters",
         ),