Skip to content

Commit 3f14cd1

Browse files
committed
chore: fix semantic versioning for pinned major or minor versions with wildcard identifier
1 parent d018442 commit 3f14cd1

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/sagemaker/jumpstart/cache.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -552,6 +552,10 @@ def _select_version(
552552
)
553553
return version_str if version_str in available_versions else None
554554

555+
if version_str[-1] == "*":
556+
# major or minor version is pinned, e.g 1.* or 1.0.*
557+
return utils.get_latest_version([version for version in available_versions if version.startswith(version_str[:-1])])
558+
555559
try:
556560
spec = SpecifierSet(f"=={version_str}")
557561
except InvalidSpecifier:

0 commit comments

Comments
 (0)