From d2e20b381bc140c90e4893b3e538e7b33f5b3ec5 Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Tue, 29 Aug 2023 15:02:51 -0700
Subject: [PATCH] Pipy validation workflow

---
 .github/scripts/validate_pipy.sh | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/.github/scripts/validate_pipy.sh b/.github/scripts/validate_pipy.sh
index 85b788cf0..8eb36098b 100644
--- a/.github/scripts/validate_pipy.sh
+++ b/.github/scripts/validate_pipy.sh
@@ -1,12 +1,19 @@
 conda create -yp ${ENV_NAME}_pypi python=${MATRIX_PYTHON_VERSION} numpy ffmpeg
+conda activate ${ENV_NAME}_pypi
 
-if [[ ${MATRIX_CHANNEL} != "release" ]]; then
-    conda run -p ${ENV_NAME}_pypi pip3 install --pre torch --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}_pypi_cudnn"
-    conda run -p ${ENV_NAME}_pypi pip3 install --pre torchvision torchaudio --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}"
+TEST_SUFFIX=""
+if [[ ${TORCH_ONLY} == 'true' ]]; then
+    TEST_SUFFIX=" --package torchonly"
+    pip3 install --pre torch --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}_pypi_cudnn"
 else
-    conda run -p ${ENV_NAME}_pypi pip3 install torch torchvision torchaudio
+    if [[ ${MATRIX_CHANNEL} != "release" ]]; then
+        pip3 install --pre torch --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}_pypi_cudnn"
+        pip3 install --pre torchvision torchaudio --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}"
+    else
+        pip3 install torch torchvision torchaudio
+    fi
 fi
 
-conda run -p ${ENV_NAME}_pypi python ./test/smoke_test/smoke_test.py
+python ./test/smoke_test/smoke_test.py ${TEST_SUFFIX} --runtime-error-check disabled
 conda deactivate
 conda env remove -p ${ENV_NAME}_pypi