diff --git a/src/ensembl/utils/database/dbconnection.py b/src/ensembl/utils/database/dbconnection.py index 2c62818..acfad51 100644 --- a/src/ensembl/utils/database/dbconnection.py +++ b/src/ensembl/utils/database/dbconnection.py @@ -157,9 +157,9 @@ def _enable_sqlite_savepoints(self, engine: sqlalchemy.engine.Engine) -> None: @event.listens_for(engine, "connect") def do_connect( - dbapi_connection: sqlalchemy.engine.interfaces.DBAPIConnection, - connection_record: sqlalchemy.pool.ConnectionPoolEntry, - ) -> None: # pylint: disable=unused-argument + dbapi_connection: Any, # SQLAlchemy is not clear about the type of this argument + connection_record: sqlalchemy.pool.ConnectionPoolEntry, # pylint: disable=unused-argument + ) -> None: """Disables emitting the BEGIN statement entirely, as well as COMMIT before any DDL.""" dbapi_connection.isolation_level = None @@ -223,8 +223,9 @@ def test_session_scope(self) -> Generator[sqlalchemy.orm.Session, None, None]: # Define a new transaction event @event.listens_for(session, "after_transaction_end") def end_savepoint( - session: sqlalchemy.orm.Session, transaction: sqlalchemy.orm.SessionTransaction - ) -> None: # pylint: disable=unused-argument + session: sqlalchemy.orm.Session, # pylint: disable=unused-argument + transaction: sqlalchemy.orm.SessionTransaction, # pylint: disable=unused-argument + ) -> None: if not connection.in_nested_transaction(): connection.begin_nested() diff --git a/tests/rloader/test_rloader.py b/tests/rloader/test_rloader.py index 24acd98..398a42d 100644 --- a/tests/rloader/test_rloader.py +++ b/tests/rloader/test_rloader.py @@ -36,8 +36,8 @@ class MockResponse: def mock_requests_get( - file_path: Path, *args: Any, **kwargs: Any -) -> MockResponse: # pylint: disable=unused-argument + file_path: Path, *args: Any, **kwargs: Any # pylint: disable=unused-argument +) -> MockResponse: """Mocks `requests.get()` function, bypassing the required internet connection.""" with file_path.open("r") as in_file: content = in_file.read()