Add CI tests for partial installations

Signed-off-by: Keval Morabia <[email protected]>

Author: kevalmorabia97

.github/workflows/partial_unit_tests.yml

@@ -0,0 +1,48 @@
+name: Partial Install Unit tests
+
+on:
+  pull_request:
+    branches: [main, release/*]
+    paths:
+      - ".github/workflows/partial_unit_tests.yml"
+      - "modelopt/**"
+      - "tests/**"
+      - "setup.py"
+      - "tox.ini"
+
+# Cancel previous runs if new commit is pushed to the same PR
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number }}
+  cancel-in-progress: true
+
+jobs:
+  partial-unit:
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+    strategy:
+      matrix:
+        test-env: [onnx, torch, torch_deploy]
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Install dependencies
+        run: pip install tox
+      - name: Run unit tests
+        run: tox -e py312-partial-unit-${{ matrix.test-env }}
+  partial-unit-windows:
+    runs-on: windows-latest
+    timeout-minutes: 30
+    strategy:
+      matrix:
+        test-env: [onnx]
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Install dependencies
+        run: pip install tox
+      - name: Run unit tests
+        run: tox -e py312-partial-unit-${{ matrix.test-env }}

tox.ini

@@ -28,7 +28,7 @@ commands =
 #####################################################################
 # Environment to run unit tests with subset of dependencies installed
 #####################################################################
-[testenv:{py39,py310,py311,py312}-ext-unit-{onnx,torch,torch_deploy}]
+[testenv:{py39,py310,py311,py312}-partial-unit-{onnx,torch,torch_deploy}]
 allowlist_externals =
     bash, rm
 deps =