Use Databricks conn_id from env for DatabricksTaskOperator in system …

…test (apache#41861)
eladkal · Aug 30, 2024 · 3740b0c · 3740b0c
1 parent 365b42f
commit 3740b0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/system/providers/databricks/example_databricks_workflow.py b/tests/system/providers/databricks/example_databricks_workflow.py
@@ -114,7 +114,7 @@
 
         task_operator_nb_1 = DatabricksTaskOperator(
             task_id="nb_1",
-            databricks_conn_id="databricks_conn",
+            databricks_conn_id=DATABRICKS_CONN_ID,
             job_cluster_key="Shared_job_cluster",
             task_config={
                 "notebook_task": {