Fix DbtCloudRunJobOperator to Use Correct Status Parameters for `re…

…use_existing_run` (apache#40048) * Pass value of `NON_TERMINAL_STATUSES` enum instead of itself * Pass `NON_TERMINAL_STATUSES` as string
romsharon98 · Jul 26, 2024 · 82ed311 · 82ed311
1 parent 7ebfe85
commit 82ed311
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/airflow/providers/dbt/cloud/operators/dbt.py b/airflow/providers/dbt/cloud/operators/dbt.py
@@ -135,7 +135,7 @@ def execute(self, context: Context):
                 account_id=self.account_id,
                 payload={
                     "job_definition_id": self.job_id,
-                    "status__in": DbtCloudJobRunStatus.NON_TERMINAL_STATUSES,
+                    "status__in": str(list(DbtCloudJobRunStatus.NON_TERMINAL_STATUSES.value)),
                     "order_by": "-created_at",
                 },
             ).json()["data"]

diff --git a/tests/providers/dbt/cloud/operators/test_dbt.py b/tests/providers/dbt/cloud/operators/test_dbt.py
@@ -374,7 +374,7 @@ def test_execute_no_wait_for_termination_and_reuse_existing_run(
             account_id=account_id,
             payload={
                 "job_definition_id": self.config["job_id"],
-                "status__in": DbtCloudJobRunStatus.NON_TERMINAL_STATUSES,
+                "status__in": str(list(DbtCloudJobRunStatus.NON_TERMINAL_STATUSES.value)),
                 "order_by": "-created_at",
             },
         )