Skip to content

Commit f7ed140

Browse files
🚧 Change cosmos to use connectio
1 parent aabeeaf commit f7ed140

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

‎dags/kids_first/dbt_bash.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,9 @@
5050
def check_for_warehouse_host_var():
5151
warehouse_host = Variable.get("INCLUDEWAREHOUSE_HOST", default=None)
5252
if warehouse_host:
53-
print("WAREHOUSE_HOST exists!")
53+
print("INCLUDEWAREHOUSE_HOST exists!")
5454
else:
55-
print("WAREHOUSE_HOST variable is not set.")
55+
print("INCLUDEWAREHOUSE_HOST variable is not set.")
5656

5757
t2 = PythonOperator(
5858
task_id="check_for_warehouse_host_var",

‎dags/kids_first/example_study.py‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,14 @@
99
ExecutionConfig,
1010
RenderConfig,
1111
)
12+
from cosmos.profiles import PostgresUserPasswordProfileMapping
1213

1314
profile_config = ProfileConfig(
1415
profile_name=Variable.get("DBT_PROFILE_NAME"),
15-
profiles_yml_filepath=Variable.get("DBT_PROFILES_YML_PATH"),
1616
target_name="prd",
17+
profile_mapping=PostgresUserPasswordProfileMapping(
18+
conn_id="postgres_dev_svc"
19+
),
1720
)
1821

1922
example_study_dag = DbtDag(

0 commit comments

Comments
 (0)