File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 14
14
gpu_arch_ver = os .getenv ("MATRIX_GPU_ARCH_VERSION" )
15
15
gpu_arch_type = os .getenv ("MATRIX_GPU_ARCH_TYPE" )
16
16
channel = os .getenv ("MATRIX_CHANNEL" )
17
- stable_version = os .getenv ("MATRIX_STABLE_VERSION" )
18
- release_version = os .getenv ("RELEASE_VERSION" )
19
17
package_type = os .getenv ("MATRIX_PACKAGE_TYPE" )
20
18
target_os = os .getenv ("TARGET_OS" )
21
19
BASE_DIR = Path (__file__ ).parent .parent .parent
@@ -69,11 +67,13 @@ def read_release_matrix():
69
67
return load_json_from_basedir ("release_matrix.json" )
70
68
71
69
def check_version (package : str ) -> None :
72
-
73
- # if release_version is specified, override stable_version coming binary matrix
70
+ release_version = os . getenv ( "RELEASE_VERSION" )
71
+ # if release_version is specified, use it to validate the packages
74
72
if (release_version ):
75
73
release_matrix = read_release_matrix ()
76
74
stable_version = release_matrix ["torch" ]
75
+ else :
76
+ stable_version = os .getenv ("MATRIX_STABLE_VERSION" )
77
77
78
78
# only makes sense to check nightly package where dates are known
79
79
if channel == "nightly" :
You can’t perform that action at this time.
0 commit comments