Add missing part to fix CI

astronomer · Jul 20, 2023 · 50a3c83 · 50a3c83
1 parent 2ef2804
commit 50a3c83
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dev/dags/example_cosmos_python_models.py b/dev/dags/example_cosmos_python_models.py
@@ -21,7 +21,7 @@
 
 DEFAULT_DBT_ROOT_PATH = Path(__file__).parent / "dbt"
 DBT_ROOT_PATH = Path(os.getenv("DBT_ROOT_PATH", DEFAULT_DBT_ROOT_PATH))
-SCHEMA = "cosmos_" + os.getenv("DATABRICKS_UNIQUE_ID", "")
+SCHEMA = "_".join("cosmos", os.getenv("DATABRICKS_UNIQUE_ID", ""))
 
 # [START example_cosmos_python_models]
 example_cosmos_python_models = DbtDag(
@@ -30,7 +30,7 @@
     dbt_project_name="jaffle_shop_python",
     conn_id="databricks_default",
     profile_args={
-        "schema": "cosmos",
+        "schema": SCHEMA,
     },
     operator_args={"append_env": True},
     profile_name_override="airflow",