Skip to content

Commit e5c3574

Browse files
committed
add helper function
1 parent a3dd208 commit e5c3574

File tree

1 file changed

+15
-2
lines changed

1 file changed

+15
-2
lines changed

src/sagemaker/fw_utils.py

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,11 +69,24 @@
6969
}
7070

7171

72+
def is_version_equal_or_higher(lowest_version, framework_version):
73+
"""Determine whether the framework_version is equal to or higher than lowest_version
74+
75+
Args:
76+
lowest_version (List[int]): lowest version represented in an integer list
77+
framework_version (str): framework version string
78+
79+
Returns:
80+
bool: Whether or not framework_version is equal to or higher than lowest_version
81+
"""
82+
version_list = [int(s) for s in framework_version.split(".")]
83+
return version_list >= lowest_version[0 : len(version_list)]
84+
85+
7286
def _is_merged_versions(framework, framework_version):
7387
lowest_version_list = MERGED_FRAMEWORKS_LOWEST_VERSIONS.get(framework)
7488
if lowest_version_list:
75-
version_list = [int(s) for s in framework_version.split(".")]
76-
return version_list >= lowest_version_list[0 : len(version_list)]
89+
return is_version_equal_or_higher(lowest_version_list, framework_version)
7790
else:
7891
return False
7992

0 commit comments

Comments
 (0)