File tree Expand file tree Collapse file tree 1 file changed +24
-0
lines changed Expand file tree Collapse file tree 1 file changed +24
-0
lines changed Original file line number Diff line number Diff line change @@ -52,6 +52,20 @@ def forward(self, x):
52
52
output = self .fc1 (x )
53
53
return output
54
54
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" )
55
69
56
70
def check_version (package : str ) -> None :
57
71
# only makes sense to check nightly package where dates are known
@@ -62,6 +76,16 @@ def check_version(package: str) -> None:
62
76
raise RuntimeError (
63
77
f"Torch version mismatch, expected { stable_version } for channel { channel } . But its { torch .__version__ } "
64
78
)
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
+
65
89
else :
66
90
print (f"Skip version check for channel { channel } as stable version is None" )
67
91
You can’t perform that action at this time.
0 commit comments