Fix: Polling should stop when terminal state is reached. (#301)

Signed-off-by: Tim Vossen <[email protected]>
Co-authored-by: Tim Vossen <[email protected]>

opensearch_py_ml/ml_commons/ml_commons_client.py

@@ -264,7 +264,7 @@ def _send_model_info(self, model_meta_json: dict):
         )
         end = time.time() + TIMEOUT  # timeout seconds
         task_flag = False
-        while not task_flag or time.time() < end:
+        while not task_flag and time.time() < end:
             time.sleep(1)
             status = self._get_task_info(output["task_id"])
             if status["state"] != "CREATED":
@@ -375,7 +375,7 @@ def get_task_info(self, task_id: str, wait_until_task_done: bool = False) -> obj
         if wait_until_task_done:
             end = time.time() + TIMEOUT  # timeout seconds
             task_flag = False
-            while not task_flag or time.time() < end:
+            while not task_flag and time.time() < end:
                 time.sleep(1)
                 output = self._get_task_info(task_id)
                 if (