Skip to content

Commit ebdbead

Browse files
committed
Fix typo
Fix version check Add not
1 parent 7b1bf7a commit ebdbead

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/smoke_test/smoke_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,8 @@ def check_version(package: str) -> None:
3636
# only makes sense to check nightly package where dates are known
3737
if channel == "nightly":
3838
check_nightly_binaries_date(options.package)
39-
else
40-
if torch.__version__ != stable_version:
39+
else:
40+
if not torch.__version__.startswith(stable_version):
4141
raise RuntimeError(
4242
f"Torch version mismatch, expected {stable_version} for channel {channel}. But its {torch.__version__}"
4343
)

0 commit comments

Comments
 (0)