Consistent pg_connection alias usage

tammy-baylis-swi · Dec 11, 2024 · e3d7941 · e3d7941
1 parent 8e8b759
commit e3d7941
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...entelemetry-instrumentation-psycopg/src/opentelemetry/instrumentation/psycopg/__init__.py b/...entelemetry-instrumentation-psycopg/src/opentelemetry/instrumentation/psycopg/__init__.py
@@ -162,7 +162,7 @@ def _instrument(self, **kwargs):
 
         dbapi.wrap_connect(
             __name__,
-            psycopg.Connection,  # pylint: disable=no-member
+            pg_connection,  # pylint: disable=no-member
             "connect",
             self._DATABASE_SYSTEM,
             self._CONNECTION_ATTRIBUTES,
@@ -189,7 +189,7 @@ def _uninstrument(self, **kwargs):
         """ "Disable Psycopg instrumentation"""
         dbapi.unwrap_connect(psycopg, "connect")  # pylint: disable=no-member
         dbapi.unwrap_connect(
-            psycopg.Connection,
+            pg_connection,
             "connect",  # pylint: disable=no-member
         )
         dbapi.unwrap_connect(