From d577e3e0ddb31bfc51a686c403a925b5a72b00bd Mon Sep 17 00:00:00 2001 From: Evan Kravitz Date: Thu, 8 Dec 2022 19:12:22 +0000 Subject: [PATCH 1/3] chore: fix failing jumpstart cache unit tests --- src/sagemaker/jumpstart/cache.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/sagemaker/jumpstart/cache.py b/src/sagemaker/jumpstart/cache.py index 202edff9ad..db607770a7 100644 --- a/src/sagemaker/jumpstart/cache.py +++ b/src/sagemaker/jumpstart/cache.py @@ -20,7 +20,7 @@ import boto3 import botocore from packaging.version import Version -from packaging.specifiers import SpecifierSet +from packaging.specifiers import SpecifierSet, InvalidSpecifier from sagemaker.jumpstart.constants import ( ENV_VARIABLE_JUMPSTART_MANIFEST_LOCAL_ROOT_DIR_OVERRIDE, ENV_VARIABLE_JUMPSTART_SPECS_LOCAL_ROOT_DIR_OVERRIDE, @@ -371,7 +371,10 @@ def _select_version( return None return str(max(available_versions)) - spec = SpecifierSet(f"=={semantic_version_str}") + try: + spec = SpecifierSet(f"=={semantic_version_str}") + except InvalidSpecifier: + raise KeyError(f"Bad semantic version: {semantic_version_str}") available_versions_filtered = list(spec.filter(available_versions)) return ( str(max(available_versions_filtered)) if available_versions_filtered != [] else None From b681d812fc03d45550a3a08f6d58970be2614cd2 Mon Sep 17 00:00:00 2001 From: Evan Kravitz Date: Thu, 8 Dec 2022 19:55:35 +0000 Subject: [PATCH 2/3] chore: pin packaging==22.0 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 4327045760..37f9af42d2 100644 --- a/setup.py +++ b/setup.py @@ -55,7 +55,7 @@ def read_requirements(filename): "protobuf3-to-dict>=0.1.5,<1.0", "smdebug_rulesconfig==1.0.1", "importlib-metadata>=1.4.0,<5.0", - "packaging>=20.0", + "packaging==22.0", "pandas", "pathos", "schema", From 216cab785cf4afccba4fa9422e070b1ec698fefe Mon Sep 17 00:00:00 2001 From: Evan Kravitz Date: Thu, 8 Dec 2022 20:09:43 +0000 Subject: [PATCH 3/3] chore: pin packaging==20.9 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 37f9af42d2..f366b147b8 100644 --- a/setup.py +++ b/setup.py @@ -55,7 +55,7 @@ def read_requirements(filename): "protobuf3-to-dict>=0.1.5,<1.0", "smdebug_rulesconfig==1.0.1", "importlib-metadata>=1.4.0,<5.0", - "packaging==22.0", + "packaging==20.9", "pandas", "pathos", "schema",