fix

Author: zewenli98

.github/workflows/linux-test.yml

@@ -114,13 +114,13 @@ jobs:
           script: ${{ inputs.pre-script }}
       - name: Download artifacts
         if: ${{ matrix.tensorrt == '' }}
-        uses: actions/download-artifact@v3
+        uses: actions/download-artifact@v4
         with:
           name: ${{ env.ARTIFACT_NAME }}
           path: /opt/torch-tensorrt-builds/
       - name: Download artifacts
         if: ${{ matrix.tensorrt != '' }}
-        uses: actions/download-artifact@v3
+        uses: actions/download-artifact@v4
         with:
           name: ${{ env.DOWNLOAD_ARTIFACT_NAME }}
           path: /opt/torch-tensorrt-builds/

.github/workflows/windows-test.yml

@@ -105,13 +105,13 @@ jobs:
           is_windows: 'enabled'
       - name: Download artifacts
         if: ${{ matrix.tensorrt == '' }}
-        uses: actions/download-artifact@v3
+        uses: actions/download-artifact@v4
         with:
           name: ${{ env.ARTIFACT_NAME }}
           path: ${{ runner.temp }}/artifacts/
       - name: Download artifacts
         if: ${{ matrix.tensorrt != '' }}
-        uses: actions/download-artifact@v3
+        uses: actions/download-artifact@v4
         with:
           name: ${{ env.DOWNLOAD_ARTIFACT_NAME }}
           path: ${{ runner.temp }}/artifacts/

noxfile.py

@@ -34,7 +34,7 @@
 # Set epochs to train VGG model for accuracy tests
 EPOCHS = 25
 
-SUPPORTED_PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"]
+SUPPORTED_PYTHON_VERSIONS = ["3.9", "3.10", "3.11", "3.12"]
 
 nox.options.sessions = [
     "l0_api_tests-" + "{}.{}".format(sys.version_info.major, sys.version_info.minor)

py/requirements.txt

@@ -1,8 +1,6 @@
 numpy
 packaging
 pybind11==2.6.2
---extra-index-url https://download.pytorch.org/whl/nightly/cu124
 torch>=2.7.0.dev,<2.8.0
 torchvision>=0.22.0.dev,<0.23.0
---extra-index-url https://pypi.ngc.nvidia.com
 pyyaml