fix(providers/amazon): fix wrong completed condition

apache · Feb 3, 2024 · ba7596a · ba7596a
1 parent 86f9c19
commit ba7596a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/airflow/providers/amazon/aws/operators/sagemaker.py b/airflow/providers/amazon/aws/operators/sagemaker.py
@@ -1104,7 +1104,7 @@ def execute(self, context: Context) -> dict:
                     {},
                     [],
                     instance_count,
-                    LogState.TAILING if job_already_completed else LogState.COMPLETE,
+                    LogState.COMPLETE if job_already_completed else LogState.TAILING,
                     description,
                     last_describe_job_call,
                 )

diff --git a/airflow/providers/amazon/aws/triggers/sagemaker.py b/airflow/providers/amazon/aws/triggers/sagemaker.py
@@ -243,7 +243,7 @@ async def run(self) -> AsyncIterator[TriggerEvent]:
         instance_count = last_description["ResourceConfig"]["InstanceCount"]
         status = last_description["TrainingJobStatus"]
         job_already_completed = status not in self.hook.non_terminal_states
-        state = LogState.TAILING if job_already_completed else LogState.COMPLETE
+        state = LogState.COMPLETE if job_already_completed else LogState.TAILING
         last_describe_job_call = time.time()
         while True:
             try: