Skip to content

Commit 00b194a

Browse files
authored
Release v1 (#1595)
* test * test
1 parent db4d5fb commit 00b194a

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

test/smoke_test/smoke_test.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,20 @@ def forward(self, x):
5252
output = self.fc1(x)
5353
return output
5454

55+
def load_json_from_basedir(filename: str):
56+
try:
57+
if os.path.exists(BASE_DIR / filename):
58+
with open(BASE_DIR / filename) as fptr:
59+
return json.load(fptr)
60+
else:
61+
return None
62+
except FileNotFoundError as exc:
63+
raise ImportError(f"File {filename} not found error: {exc.strerror}") from exc
64+
except json.JSONDecodeError as exc:
65+
raise ImportError(f"Invalid JSON {filename}") from exc
66+
67+
def read_release_matrix():
68+
return load_json_from_basedir("release_matrix.json")
5569

5670
def check_version(package: str) -> None:
5771
# only makes sense to check nightly package where dates are known
@@ -62,6 +76,16 @@ def check_version(package: str) -> None:
6276
raise RuntimeError(
6377
f"Torch version mismatch, expected {stable_version} for channel {channel}. But its {torch.__version__}"
6478
)
79+
release_version = read_release_matrix()
80+
if package == "all":
81+
for module in MODULES:
82+
imported_module = importlib.import_module(module["name"])
83+
module_version = imported_module.__version__
84+
if not module_version.startswith(release_version[module["name"]]):
85+
raise RuntimeError(
86+
f"{module['name']} version mismatch, expected {release_version[module['name']]} for channel {channel}. But its {module_version}"
87+
)
88+
6589
else:
6690
print(f"Skip version check for channel {channel} as stable version is None")
6791

0 commit comments

Comments
 (0)