Add another note

tanmayv25 · tanmayv25 · commit 21ae4f86571c · 2025-06-06T14:58:38.000-07:00
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -33,12 +33,15 @@
 
 
 def pytest_collection_modifyitems(config, items):
-    # Handle tests marked with tensorrtllm
+    """
+    This function is called to modify the list of tests to run.
+    It is used to skip tests that are not supported on all environments.
+    """
+
+    # Tests marked with tensorrtllm requires specific environment with tensorrtllm
+    # installed. Hence, we skip them if the user did not explicitly ask for them.
     if config.getoption("-m") and "tensorrtllm" in config.getoption("-m"):
-        # User explicitly asked for tensorrtllm tests, do nothing
         return
-
-    # Otherwise, skip all tests marked with tensorrtllm
     skip_tensorrtllm = pytest.mark.skip(reason="need -m tensorrtllm to run")
     for item in items:
         if "tensorrtllm" in item.keywords:
