clen up

Arize-ai · Sep 17, 2024 · 2986714 · 2986714
1 parent 4885b4b
commit 2986714
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integration_tests/_helpers.py b/integration_tests/_helpers.py
@@ -624,7 +624,7 @@ def _random_schema(
 ) -> Iterator[str]:
     engine = create_engine(url.set(drivername="postgresql+psycopg"))
     try:
-        engine.connect()
+        engine.connect().close()
     except OperationalError as exc:
         if "too many clients" in str(exc):
             pass