fix: warm up slack commands

ZettaAI · Jan 23, 2023 · 011e525 · 011e525
1 parent 891553c
commit 011e525
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/slackbot/airflow_api.py b/slackbot/airflow_api.py
@@ -4,6 +4,7 @@
 import concurrent.futures
 
 from airflow import settings
+from airflow.hooks.base import BaseHook
 from airflow.models import DagBag, DagModel, Variable, Connection
 from airflow.models.dagrun import DagRun, DagRunType
 from airflow.api.common.mark_tasks import set_dag_run_state_to_success

diff --git a/slackbot/warm_up_commands.py b/slackbot/warm_up_commands.py
@@ -10,7 +10,7 @@
     description="Keep at least 10 nodes running in a cluster for 1 hour",
     exclusive=True,  # updates "task owner" for slack responses
     cancelable=False,
-    extra_parameters=False,
+    extra_parameters=True,
 )
 def warm_up(msg: dict) -> None:
     """Sets the min size variable for a cluster and runs cluster_management."""