add skip for generate_dbt_docs_gcs

astronomer · Oct 24, 2023 · 1e9f33b · 1e9f33b
1 parent 750eb6d
commit 1e9f33b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dev/dags/dbt_docs.py b/dev/dags/dbt_docs.py
@@ -58,6 +58,11 @@ def which_upload():
         downstream_tasks_to_run += ["generate_dbt_docs_azure"]
     except AirflowNotFoundException:
         pass
+    try:
+        BaseHook.get_connection(GCS_CONN_ID)
+        downstream_tasks_to_run += ["generate_dbt_docs_gcs"]
+    except AirflowNotFoundException:
+        pass
 
     return downstream_tasks_to_run
 
@@ -94,4 +99,4 @@ def which_upload():
         bucket_name="cosmos-docs",
     )
 
-    which_upload() >> [generate_dbt_docs_aws, generate_dbt_docs_azure]
+    which_upload() >> [generate_dbt_docs_aws, generate_dbt_docs_azure, generate_dbt_docs_gcs]