test: Add sanity test for vllm.distributed patched modules

container/deps/vllm/data_plane/kv_cache.py

@@ -16,6 +16,9 @@
 # A script to download a Python wheel, patch it, copy additional files,
 # repackage it, and optionally install the new wheel.
 
+# FIXME: Address type checking with divergent interfaces for Ucp/Nccl data planes
+# type: ignore
+
 import typing
 
 if typing.TYPE_CHECKING:

container/deps/vllm/tests/test_patch_install.py

@@ -18,4 +18,8 @@ def test_version():
 @pytest.mark.skipif(vllm is None, reason="Skipping vllm tests, vllm not installed")
 def test_patch_imports():
     # Verify patched files have no glaring syntax or import issues
-    pass
+    import vllm.distributed.data_plane as d
+    import vllm.distributed.kv_cache as k
+
+    # Placeholder to avoid unused import errors or removal by linters
+    assert d, k

pyproject.toml

@@ -92,3 +92,8 @@ exclude = ["model.py"]
 ignore_missing_imports = true
 
 check_untyped_defs = true
+
+[[tool.mypy.overrides]]
+# Skip mypy analysis on internal dependencies of vllm
+module = ["vllm.*"]
+follow_imports = "skip"