suppress pyright false positives

phelps-sg · Nov 7, 2023 · 8e145b6 · 8e145b6
1 parent 656a219
commit 8e145b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/openai_pygenerator/openai_pygenerator.py b/src/openai_pygenerator/openai_pygenerator.py
@@ -117,10 +117,10 @@ def generate_completions(
             request_timeout=request_timeout,
         )
         logger.debug("response = %s", result)
-        for choice in result.choices:
+        for choice in result.choices:  # type: ignore
             yield choice.message
     except (
-        openai.error.Timeout,
+        openai.error.Timeout,  # type: ignore
         urllib3.exceptions.TimeoutError,
         RateLimitError,
         APIConnectionError,