Skip to content

Commit 54da893

Browse files
authored
Fix nightly smoke test (#1340)
* Fix nightly smoke test * Fix nightly builds
1 parent 2b968b7 commit 54da893

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

.github/scripts/validate_binaries.sh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ else
2020

2121

2222
# Special case Pypi installation package, only applicable to linux nightly CUDA 11.7 builds, wheel package
23-
if [[ ${TARGET_OS} == 'linux' && ${MATRIX_GPU_ARCH_VERSION} == '11.7' && ${MATRIX_PACKAGE_TYPE} == 'manywheel' ]]; then
23+
if [[ ${TARGET_OS} == 'linux' && ${MATRIX_GPU_ARCH_VERSION} == '11.7' && ${MATRIX_PACKAGE_TYPE} == 'manywheel' && ${MATRIX_CHANNEL} != 'nightly' ]]; then
2424
conda create -yp ${ENV_NAME}_pypi python=${MATRIX_PYTHON_VERSION} numpy ffmpeg
2525
INSTALLATION_PYPI=${MATRIX_INSTALLATION/"cu117"/"cu117_pypi_cudnn"}
2626
INSTALLATION_PYPI=${INSTALLATION_PYPI/"torchvision torchaudio"/""}
@@ -35,7 +35,6 @@ else
3535
conda create -y -n ${ENV_NAME} python=${MATRIX_PYTHON_VERSION} numpy ffmpeg
3636
conda activate ${ENV_NAME}
3737
INSTALLATION=${MATRIX_INSTALLATION/"conda install"/"conda install -y"}
38-
INSTALLATION=${INSTALLATION/"extra-index-url"/"index-url"}
3938
eval $INSTALLATION
4039

4140
if [[ ${TARGET_OS} == 'linux' ]]; then

test/smoke_test/smoke_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
def check_version(package: str) -> None:
3737
# only makes sense to check nightly package where dates are known
3838
if channel == "nightly":
39-
check_nightly_binaries_date(options.package)
39+
check_nightly_binaries_date(package)
4040
else:
4141
if not torch.__version__.startswith(stable_version):
4242
raise RuntimeError(

0 commit comments

Comments
 (0)