Revert "Consistent pg_connection alias usage"

This reverts commit e3d7941.
tammy-baylis-swi · Dec 11, 2024 · a94a591 · a94a591
1 parent e3d7941
commit a94a591
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__,
-            pg_connection,  # pylint: disable=no-member
+            psycopg.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(
-            pg_connection,
+            psycopg.Connection,
             "connect",  # pylint: disable=no-member
         )
         dbapi.unwrap_connect(