Fix version check

atalman · atalman · commit 65a4b5dddce1 · 2023-03-06T14:43:34.000-08:00
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -37,7 +37,7 @@ def check_version(package: str) -> None:
     if channel == "nightly":
         check_nightly_binaries_date(options.package)
     else:
-        if torch.__version__ != stable_version:
+        if torch.__version__.startswith(stable_version):
             raise RuntimeError(
                 f"Torch version mismatch, expected {stable_version} for channel {channel}. But its {torch.__version__}"
             )

Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@ def check_version(package: str) -> None:`
`37`	`37`	`if channel == "nightly":`
`38`	`38`	`check_nightly_binaries_date(options.package)`
`39`	`39`	`else:`
`40`		`- if torch.__version__ != stable_version:`
	`40`	`+ if torch.__version__.startswith(stable_version):`
`41`	`41`	`raise RuntimeError(`
`42`	`42`	`f"Torch version mismatch, expected {stable_version} for channel {channel}. But its {torch.__version__}"`
`43`	`43`	`)`