From 50a3c8333bf8f14a1481c26f1c072f44b2c8e0a6 Mon Sep 17 00:00:00 2001
From: Tatiana Al-Chueyr <tatiana.alchueyr@gmail.com>
Date: Thu, 20 Jul 2023 10:24:46 +0100
Subject: [PATCH] Add missing part to fix CI

---
 dev/dags/example_cosmos_python_models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dev/dags/example_cosmos_python_models.py b/dev/dags/example_cosmos_python_models.py
index d8083c518..cbd0ad0d2 100644
--- 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",