def _get_dependency_upper_bound_for_semver()

in sagemaker_image_builder/dependency_upgrader.py [0:0]


def _get_dependency_upper_bound_for_semver(lower_bound: str, runtime_upgrade_type):
    lower_semver = Version.parse(lower_bound, optional_minor_and_patch=True)
    if runtime_upgrade_type == _MAJOR:
        return ""  # No upper bound.
    elif runtime_upgrade_type == _MINOR:
        return f",<{lower_semver.bump_major()}"
    elif runtime_upgrade_type == _PATCH:
        return f",<{lower_semver.bump_minor()}"
    else:
        raise Exception()