finally workin

mpkrass7 · Jun 24, 2024 · c7ca042 · c7ca042
1 parent e8dd6b3
commit c7ca042
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 5 deletions.
diff --git a/{{ cookiecutter.repo_name }}/conf/base/parameters.yml b/{{ cookiecutter.repo_name }}/conf/base/parameters.yml
@@ -54,6 +54,21 @@ deploy_forecast:
     label: {{ cookiecutter.project_name }}
     description: A fantastically generated forecast from Kedro, hurray!
     prediction_interval: 80
+  retraining_policy:
+    autopilotOptions:
+      blendBestModels: false
+      mode: quick
+      shapOnlyMode: false
+    description: Retrain monthly and add best model to challengers
+    name: Retrain Monthly
+    trigger:
+      schedule:
+        dayOfMonth: [1]
+        hour: [0]
+        minute: [0]
+      statusDeclinesToFailing: false
+      statusDeclinesToWarning: false
+      statusStillInDecline: false
 
 deploy_streamlit_app:
   scoring_dataset_name: {{ cookiecutter.project_name }} Scoring Data

diff --git a/...ter.repo_name }}/src/{{ cookiecutter.python_package }}/pipelines/deploy_forecast/nodes.py b/...ter.repo_name }}/src/{{ cookiecutter.python_package }}/pipelines/deploy_forecast/nodes.py
@@ -8,7 +8,7 @@
 from __future__ import annotations #Keep at top of file
 
 import time
-from typing import Any, List, TYPE_CHECKING, Union, Optional
+from typing import Any, List, TYPE_CHECKING, Union, Optional, Dict
 
 from datarobot.models.use_cases.utils import UseCaseLike
 from datarobot import Dataset 
@@ -158,10 +158,10 @@ def ensure_deployment_settings(
 
     user_id = deployment.owners["preview"][0]["id"]  # type: ignore
 
-    # client.patch(f"deployments/{deployment_id}/settings",
-    #              json={
-    #                  "automaticActuals": {"enabled": True}
-    #              })
+    client.patch(f"deployments/{deployment_id}/settings",
+                 json={
+                     "automaticActuals": {"enabled": True}
+                 })
 
     # set up retraining
     try:
@@ -186,3 +186,4 @@ def ensure_deployment_settings(
             "retrainingUserId": user_id,
         },
     )
+