move tensorrt to a dedicated requirement file.

Author: GdoongMathew

.github/workflows/ci-tests-pytorch.yml

@@ -139,6 +139,7 @@ jobs:
           pip install ".[${EXTRA_PREFIX}extra,${EXTRA_PREFIX}test,${EXTRA_PREFIX}strategies]" \
             -U --upgrade-strategy=eager --prefer-binary \
             -r requirements/_integrations/accelerators.txt \
+            -r requirements/_integrations/tensorrt.txt \
             --extra-index-url="${TORCH_URL}" --find-links="${PYPI_CACHE_DIR}"
           pip list
       - name: Drop LAI from extensions

requirements/_integrations/tensorrt.txt

@@ -0,0 +1,2 @@
+--extra-index-url https://download.pytorch.org/whl/torch-tensorrt
+torch-tensorrt >= 2.3.0; platform_system == "Linux" and python_version >= "3.12"  # Limit the torch-tensorrt version for testing.

requirements/pytorch/test.txt

@@ -18,6 +18,3 @@ fastapi  # for `ServableModuleValidator`  # not setting version as re-defined in
 uvicorn  # for `ServableModuleValidator`  # not setting version as re-defined in App
 
 tensorboard >=2.9.1, <2.21.0  # for `TensorBoardLogger`
-
---extra-index-url https://download.pytorch.org/whl/torch-tensorrt
-torch-tensorrt >= 2.3.0; platform_system == "Linux" and python_version >= "3.12"  # Limit the torch-tensorrt version for testing.