Skip to content

Commit a94a591

Browse files
Revert "Consistent pg_connection alias usage"
This reverts commit e3d7941.
1 parent e3d7941 commit a94a591

File tree

1 file changed

+2
-2
lines changed
  • instrumentation/opentelemetry-instrumentation-psycopg/src/opentelemetry/instrumentation/psycopg

1 file changed

+2
-2
lines changed

instrumentation/opentelemetry-instrumentation-psycopg/src/opentelemetry/instrumentation/psycopg/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,7 @@ def _instrument(self, **kwargs):
162162

163163
dbapi.wrap_connect(
164164
__name__,
165-
pg_connection, # pylint: disable=no-member
165+
psycopg.Connection, # pylint: disable=no-member
166166
"connect",
167167
self._DATABASE_SYSTEM,
168168
self._CONNECTION_ATTRIBUTES,
@@ -189,7 +189,7 @@ def _uninstrument(self, **kwargs):
189189
""" "Disable Psycopg instrumentation"""
190190
dbapi.unwrap_connect(psycopg, "connect") # pylint: disable=no-member
191191
dbapi.unwrap_connect(
192-
pg_connection,
192+
psycopg.Connection,
193193
"connect", # pylint: disable=no-member
194194
)
195195
dbapi.unwrap_connect(

0 commit comments

Comments
 (0)