Pass is_singular_airflow_task env var

stellar · Sep 18, 2024 · 94e4174 · 94e4174
1 parent 7ee804b
commit 94e4174
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/dags/dbt_singular_tests_dag.py b/dags/dbt_singular_tests_dag.py
@@ -34,6 +34,7 @@
         command_type="test",
         tag="singular_test",
         resource_cfg="dbt",
+        run_singular_test="true"
     )
 
     singular_tests
diff --git a/dags/stellar_etl_airflow/build_dbt_task.py b/dags/stellar_etl_airflow/build_dbt_task.py
@@ -67,6 +67,7 @@ def dbt_task(
     command_type="build",
     excluded=None,
     resource_cfg="default",
+    run_singular_test="false"
 ):
     namespace = conf.get("kubernetes", "NAMESPACE")
     if namespace == "default":
@@ -134,6 +135,7 @@ def dbt_task(
             "PUBLIC_SOURCE_SCHEMA": "{{ var.value.dbt_public_source_schema }}",
             "EXECUTION_DATE": "{{ ts }}",
             "AIRFLOW_START_TIMESTAMP": "{{ ti.start_date.strftime('%Y-%m-%dT%H:%M:%SZ') }}",
+            "IS_SINGULAR_AIRFLOW_TASK": run_singular_test,
         },
         image=dbt_image,
         arguments=args,