pytorch · lanluo-nvidia · Jun 4, 2024 · May 14, 2024 · Jun 4, 2024 · peri044
diff --git a/packaging/pre_build_script_windows.sh b/packaging/pre_build_script_windows.sh
@@ -6,3 +6,5 @@ python -m pip install tensorrt==10.0.1 --extra-index-url https://pypi.nvidia.com
 choco install bazelisk -y
 
 cat toolchains/ci_workspaces/WORKSPACE.win.release.tmpl | envsubst > WORKSPACE
+
+echo "RELEASE=1" >> ${GITHUB_ENV}
diff --git a/tests/py/requirements.txt b/tests/py/requirements.txt
@@ -12,4 +12,5 @@ tensorrt==10.0.1
 timm>=1.0.3
 transformers==4.39.3
 parameterized>=0.2.0
-expecttest==0.1.6
+expecttest==0.1.6
+numpy==1.24.1
Original file line number	Diff line number	Diff line change
		@@ -6,3 +6,5 @@ python -m pip install tensorrt==10.0.1 --extra-index-url https://pypi.nvidia.com
		choco install bazelisk -y

		cat toolchains/ci_workspaces/WORKSPACE.win.release.tmpl \| envsubst > WORKSPACE

		echo "RELEASE=1" >> ${GITHUB_ENV}
Copy link Collaborator peri044 May 17, 2024 Choose a reason for hiding this comment The reason will be displayed to describe this comment to others. Learn more. How does this related to channel=release in the yml files ? Copy link Collaborator Author gs-olive May 17, 2024 Choose a reason for hiding this comment The reason will be displayed to describe this comment to others. Learn more. Based on the discussion here: #2806 (comment), this flag seems to get the package naming correct on the Torch upload index. I do not think it directly relates to the `channel` from the `.yml` files.