Skip to content

Commit 953c5a1

Browse files
fix lint
1 parent 36b0589 commit 953c5a1

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

drift/instrumentation/psycopg2/instrumentation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ def cursor(self, name: str | None = None, cursor_factory: Any = None, *args: Any
219219
self._sdk,
220220
cursor_factory, # This becomes the base class (None uses default)
221221
)
222-
return self._connection.cursor(name=name, cursor_factory=wrapped_factory, *args, **kwargs)
222+
return self._connection.cursor(*args, name=name, cursor_factory=wrapped_factory, **kwargs)
223223

224224
def __getattr__(self, name: str) -> Any:
225225
"""Proxy all other methods/attributes to the real connection."""
@@ -852,7 +852,7 @@ def _mock_execute_with_data(self, cursor: Any, mock_data: dict[str, Any]) -> Non
852852
# If it's a dict cursor and we have description, convert rows to dicts
853853
if is_dict_cursor and description_data:
854854
column_names = [col["name"] for col in description_data]
855-
mock_rows = [dict(zip(column_names, row)) for row in mock_rows]
855+
mock_rows = [dict(zip(column_names, row, strict=True)) for row in mock_rows]
856856

857857
cursor._mock_rows = mock_rows # pyright: ignore[reportAttributeAccessIssue]
858858
cursor._mock_index = 0 # pyright: ignore[reportAttributeAccessIssue]

0 commit comments

Comments
 (0)