refactor: use yield from in connections attempts

psycopg · Dec 1, 2023 · be554d7 · be554d7
1 parent 7dc5852
commit be554d7
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/psycopg/psycopg/conninfo.py b/psycopg/psycopg/conninfo.py
@@ -294,11 +294,9 @@ def conninfo_attempts(params: ConnDict) -> Iterator[ConnDict]:
     if params.get("load_balance_hosts", "disable") == "random":
         attempts = list(_split_attempts(_inject_defaults(params)))
         shuffle(attempts)
-        for attempt in attempts:
-            yield attempt
+        yield from attempts
     else:
-        for attempt in _split_attempts(_inject_defaults(params)):
-            yield attempt
+        yield from _split_attempts(_inject_defaults(params))
 
 
 async def conninfo_attempts_async(params: ConnDict) -> AsyncIterator[ConnDict]: