Skip to content

Commit e3d7941

Browse files
Consistent pg_connection alias usage
1 parent 8e8b759 commit e3d7941

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-
psycopg.Connection, # pylint: disable=no-member
165+
pg_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-
psycopg.Connection,
192+
pg_connection,
193193
"connect", # pylint: disable=no-member
194194
)
195195
dbapi.unwrap_connect(

0 commit comments

Comments
 (0)