Refactor DatabricksOrchestrator environment variable handling

zenml-io · Jul 11, 2024 · 66c08c7 · 66c08c7
1 parent f69233e
commit 66c08c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/zenml/integrations/databricks/orchestrators/databricks_orchestrator_entrypoint_config.py b/src/zenml/integrations/databricks/orchestrators/databricks_orchestrator_entrypoint_config.py
@@ -24,7 +24,7 @@
 )
 
 WHEEL_PACKAGE_OPTION = "wheel_package"
-DATABRICKS_JOB_ID_OPTION = "job_id"
+DATABRICKS_JOB_ID_OPTION = "databricks_job_id"
 ENV_ZENML_DATABRICKS_ORCHESTRATOR_RUN_ID = (
     "ZENML_DATABRICKS_ORCHESTRATOR_RUN_ID"
 )