Pass JOBS_API_VERSION to databricks sdk calls

astronomer · Mar 19, 2024 · ed147a6 · ed147a6
1 parent 11f143b
commit ed147a6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/astro_databricks/operators/workflow.py b/src/astro_databricks/operators/workflow.py
@@ -4,8 +4,6 @@
 from logging import Logger
 from typing import TYPE_CHECKING
 
-from constants import JOBS_API_VERSION
-
 if TYPE_CHECKING:
     pass
 import json
@@ -28,6 +26,7 @@
 from databricks_cli.sdk.api_client import ApiClient
 from mergedeep import merge
 
+from astro_databricks.constants import JOBS_API_VERSION
 from astro_databricks.plugins.plugin import (
     DatabricksJobRepairAllFailedLink,
     DatabricksJobRunLink,