Skip to content

Commit ef74ced

Browse files
committed
Make smoke-test runnable without envvars
1 parent 410ec8e commit ef74ced

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/smoke_test/smoke_test.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,11 +57,13 @@ def check_version(package: str) -> None:
5757
# only makes sense to check nightly package where dates are known
5858
if channel == "nightly":
5959
check_nightly_binaries_date(package)
60-
else:
60+
elif stable_version is not None:
6161
if not torch.__version__.startswith(stable_version):
6262
raise RuntimeError(
6363
f"Torch version mismatch, expected {stable_version} for channel {channel}. But its {torch.__version__}"
6464
)
65+
else:
66+
print(f"Skip version check for channel {channel} as stable version is None")
6567

6668
def check_nightly_binaries_date(package: str) -> None:
6769
from datetime import datetime, timedelta

0 commit comments

Comments
 (0)