From b23e3e92714e4d5326ee153fa7a96afb62df3ad8 Mon Sep 17 00:00:00 2001
From: Vinit Janbandhu <Vinit.Janbandhu@ibm.com>
Date: Mon, 5 Oct 2026 18:34:00 +0530
Subject: [PATCH] Set BUILD_VERSION and PYTORCH_VERSION environment variables

Pin the version string for the torchvision wheel name.
---
 t/torchvision/torchvision_0.29.0_ubi_10.2.sh | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/t/torchvision/torchvision_0.29.0_ubi_10.2.sh b/t/torchvision/torchvision_0.29.0_ubi_10.2.sh
index c916262dee..390b9be973 100644
--- a/t/torchvision/torchvision_0.29.0_ubi_10.2.sh
+++ b/t/torchvision/torchvision_0.29.0_ubi_10.2.sh
@@ -147,6 +147,9 @@ PYEOF
 # ---------------------------------------------------------------------------
 export FORCE_CUDA=0
 export TORCHVISION_USE_NVJPEG=0
+# Pin the version string so the wheel name is torchvision-0.29.0-* instead of torchvision-0.29.0a0+<githash>-*
+export BUILD_VERSION="${PACKAGE_VERSION#v}"
+export PYTORCH_VERSION="${TORCH_VERSION}"
 
 if ! python3.14 -m pip wheel \
         --no-build-isolation \