fix lint

Author: sohankshirsagar

drift/core/drift_sdk.py

@@ -443,7 +443,7 @@ def _init_auto_instrumentations(self) -> None:
             pass
 
         try:
-            import httpx  # type: ignore[unresolved-import]
+            import httpx
 
             from ..instrumentation.httpx import HttpxInstrumentation
 
@@ -473,7 +473,7 @@ def _init_auto_instrumentations(self) -> None:
             pass
 
         try:
-            import sqlalchemy  # type: ignore[unresolved-import]
+            import sqlalchemy
 
             from ..instrumentation.sqlalchemy import SqlAlchemyInstrumentation
 
@@ -490,7 +490,7 @@ def _init_auto_instrumentations(self) -> None:
 
         # Try psycopg2 first
         try:
-            import psycopg2  # type: ignore[unresolved-import]
+            import psycopg2
 
             from ..instrumentation.psycopg2 import Psycopg2Instrumentation
 
@@ -502,7 +502,7 @@ def _init_auto_instrumentations(self) -> None:
 
         # Try psycopg (v3)
         try:
-            import psycopg  # type: ignore[unresolved-import]
+            import psycopg
 
             from ..instrumentation.psycopg import PsycopgInstrumentation
 
@@ -518,7 +518,7 @@ def _init_auto_instrumentations(self) -> None:
             logger.debug("Both psycopg2 and psycopg available - instrumented both")
 
         try:
-            import redis  # type: ignore[unresolved-import]
+            import redis
 
             from ..instrumentation.redis import RedisInstrumentation
 
@@ -528,7 +528,7 @@ def _init_auto_instrumentations(self) -> None:
             pass
 
         try:
-            import grpc  # type: ignore[unresolved-import]
+            import grpc
 
             from ..instrumentation.grpc import GrpcInstrumentation
 

drift/instrumentation/django/instrumentation.py

@@ -130,7 +130,7 @@ def patched_setup(*args, **kwargs):
             finally:
                 django.setup = original_setup
 
-        django.setup = patched_setup
+        django.setup = patched_setup  # ty: ignore[invalid-assignment]
         logger.debug("Deferred middleware injection to django.setup()")
 
     def _force_database_reconnect(self) -> None:

drift/instrumentation/urllib3/e2e-tests/src/app.py

@@ -548,8 +548,7 @@ def test_preload_content_false_gzip():
 
 @app.route("/test/preload-false-decode-content-false-gzip", methods=["GET"])
 def test_preload_false_decode_content_false_gzip():
-    """Test preload_content=False + decode_content=False with gzip response.
-    """
+    """Test preload_content=False + decode_content=False with gzip response."""
     import gzip as gzip_mod
 
     try:
@@ -570,20 +569,24 @@ def test_preload_false_decode_content_false_gzip():
         try:
             decompressed = gzip_mod.decompress(raw_bytes)
             data = json.loads(decompressed.decode("utf-8"))
-            return jsonify({
-                "raw_bytes_length": len(raw_bytes),
-                "decompressed_length": len(decompressed),
-                "was_compressed": len(raw_bytes) != len(decompressed),
-                "data": data,
-            })
+            return jsonify(
+                {
+                    "raw_bytes_length": len(raw_bytes),
+                    "decompressed_length": len(decompressed),
+                    "was_compressed": len(raw_bytes) != len(decompressed),
+                    "data": data,
+                }
+            )
         except gzip_mod.BadGzipFile:
             # If we get here during replay, it means the bytes were already
             # decompressed - this is the bug
-            return jsonify({
-                "error": "BadGzipFile - bytes were not compressed as expected",
-                "raw_bytes_length": len(raw_bytes),
-                "raw_bytes_preview": raw_bytes[:100].decode("utf-8", errors="replace"),
-            }), 500
+            return jsonify(
+                {
+                    "error": "BadGzipFile - bytes were not compressed as expected",
+                    "raw_bytes_length": len(raw_bytes),
+                    "raw_bytes_preview": raw_bytes[:100].decode("utf-8", errors="replace"),
+                }
+            ), 500
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 

pyproject.toml

@@ -133,6 +133,7 @@ exclude = ["**/e2e-tests/**", "**/stack-tests/**"]
 [[tool.ty.overrides]]
 # Disable unresolved-import errors for instrumentation files with optional dependencies
 include = [
+    "drift/core/**",
     "drift/instrumentation/django/**",
     "drift/instrumentation/grpc/**",
     "drift/instrumentation/psycopg/**",
@@ -141,6 +142,7 @@ include = [
     "drift/instrumentation/pyjwt/**",
     "drift/instrumentation/redis/**",
     "drift/instrumentation/kinde/**",
+    "drift/instrumentation/urllib3/**",
     "drift/instrumentation/http/transform_engine.py",
 ]