fix type warnings

Author: podocarp

drift/instrumentation/e2e_common/base_runner.py

@@ -543,6 +543,7 @@ def _print_comparison(self, baseline: dict[str, dict], sdk: dict[str, dict]):
             cur = sdk.get(name)
 
             if base is None:
+                assert cur is not None
                 self.log(f"{name:<40} {'N/A':>12} {cur['ops']:>10.2f}/s {'':>10}")
                 continue
             if cur is None:

drift/instrumentation/psycopg/instrumentation.py

@@ -243,7 +243,7 @@ async def patched_async_connect(conninfo="", **kwargs):
         # Replace the classmethod with our patched version
         AsyncConnection.connect = classmethod(
             lambda cls, conninfo="", **kwargs: patched_async_connect(conninfo, **kwargs)
-        )  # type: ignore[method-assign]
+        )
         logger.info("psycopg.AsyncConnection.connect instrumented")
 
         # Also patch AsyncConnectionPool to inject cursor_factory
@@ -281,7 +281,7 @@ def patched_init(pool_self, conninfo="", **kwargs):
 
             return original_init(pool_self, conninfo, **kwargs)
 
-        AsyncConnectionPool.__init__ = patched_init  # type: ignore[method-assign]
+        AsyncConnectionPool.__init__ = patched_init
         logger.info("psycopg_pool.AsyncConnectionPool.__init__ instrumented")
 
     def _create_async_cursor_factory(self, sdk: TuskDrift, base_factory=None):
@@ -296,7 +296,7 @@ def _create_async_cursor_factory(self, sdk: TuskDrift, base_factory=None):
             from psycopg import AsyncCursor as BaseAsyncCursor
         except ImportError:
             logger.warning("[ASYNC_CURSOR_FACTORY] Could not import psycopg.AsyncCursor")
-            BaseAsyncCursor = object  # type: ignore
+            BaseAsyncCursor = object
 
         base = base_factory or BaseAsyncCursor