From a986a0f839ce0d8b8447da96ef43c595d0aea2b4 Mon Sep 17 00:00:00 2001
From: John Eastman <john.eastman@datarobot.com>
Date: Tue, 23 Jul 2024 15:56:42 -0400
Subject: [PATCH] Minor bug fixes for streamlit app

---
 YoutubeForecastMaker/conf/base/parameters.yml      | 14 +++++++-------
 .../pipelines/deploy_forecast/nodes.py             |  1 +
 .../pipelines/deploy_streamlit_app/pipeline.py     |  2 +-
 3 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/YoutubeForecastMaker/conf/base/parameters.yml b/YoutubeForecastMaker/conf/base/parameters.yml
index db8d495..f2c8a8a 100644
--- 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
index 65884f2..5bf242d 100644
--- 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
index e13de63..d89d65d 100644
--- 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",
         ),