remove unncessary stopiteration errors in generator function

kiwix · Aug 7, 2024 · 95c2609 · 95c2609
1 parent 1dfd89a
commit 95c2609
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/worker/manager/src/mirrors_qa_manager/worker.py b/worker/manager/src/mirrors_qa_manager/worker.py
@@ -300,7 +300,7 @@ def fetch_tests(self) -> Iterable[dict[str, str]]:
             data = self.query_api(f"/tests?{params}")
             nb_tests = data["metadata"]["page_size"]
             if nb_tests == 0:  # No more pending tests to fetch
-                raise StopIteration
+                break
 
             logger.info(f"Fetched {nb_tests} test(s) from Backend API")
 
@@ -314,7 +314,7 @@ def fetch_tests(self) -> Iterable[dict[str, str]]:
             yield from data["tests"]
 
             if current_page == last_page:
-                raise StopIteration
+                break
 
     def sleep(self) -> None:
         logger.info(f"Sleeping for {Settings.SLEEP_SECONDS}s")