Clean up code changes

.github/scripts/install-torch-tensorrt-windows.sh

@@ -1,9 +1,14 @@
 set -eou pipefail
 source "${BUILD_ENV_FILE}"
-# ${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision
+
+# Install test index version of Torch and Torchvision
 ${CONDA_RUN} pip install torch torchvision --index-url https://download.pytorch.org/whl/test/${CU_VERSION}
 ${CONDA_RUN} pip install pyyaml mpmath==1.3.0
+
+# Install TRT 10 from PyPi
 ${CONDA_RUN} pip install tensorrt==10.0.0b6 tensorrt-${CU_VERSION::4}-bindings==10.0.0b6 tensorrt-${CU_VERSION::4}-libs==10.0.0b6 --extra-index-url https://pypi.nvidia.com
+
+# Install pre-built Torch-TRT
 ${CONDA_RUN} pip install ${RUNNER_ARTIFACT_DIR}/torch_tensorrt*.whl
 
 echo -e "Running test script";

packaging/pre_build_script_windows.sh

@@ -1,5 +1,9 @@
 python -m pip install -U numpy packaging pyyaml setuptools wheel
+
+# Install Torch from Test Index
 python -m pip install torch==2.3.0 --index-url https://download.pytorch.org/whl/test/${CU_VERSION}
+
+# Install TRT 10 from PyPi
 python -m pip install tensorrt==10.0.0b6 tensorrt-${CU_VERSION::4}-bindings==10.0.0b6 tensorrt-${CU_VERSION::4}-libs==10.0.0b6 --extra-index-url https://pypi.nvidia.com
 
 choco install bazelisk -y

setup.py

@@ -82,7 +82,8 @@ def load_dep_info():
 
 CXX11_ABI = False
 JETPACK_VERSION = None
-PY_ONLY = sys.platform.startswith("win")
+# TODO: Remove once C++ Runtime is integrated in Windows
+PY_ONLY = IS_WINDOWS
 NO_TS = False
 LEGACY = False
 RELEASE = False