fix torchvision unit test

Ruinong Tian · Ruinong Tian · commit cac37b079b07 · 2025-06-04T19:10:32.000Z
diff --git a/test/test_artifacts/v2/scripts/run_torchvision_tests.sh b/test/test_artifacts/v2/scripts/run_torchvision_tests.sh
@@ -2,7 +2,7 @@
 
 set -e 
 
-torchvision_version=$(micromamba list | grep torchvision | tr -s ' ' | cut -d ' ' -f 3)
+torchvision_version=$(micromamba list | awk '$1=="torchvision" {print $2}')
 
 git checkout tags/v$torchvision_version
 
diff --git a/test/test_artifacts/v3/scripts/run_torchvision_tests.sh b/test/test_artifacts/v3/scripts/run_torchvision_tests.sh
@@ -2,7 +2,7 @@
 
 set -e 
 
-torchvision_version=$(micromamba list | grep torchvision | tr -s ' ' | cut -d ' ' -f 3)
+torchvision_version=$(micromamba list | awk '$1=="torchvision" {print $2}')
 
 git checkout tags/v$torchvision_version
 
