fix: JumpStartModelsAccessor

Author: evakravi

src/sagemaker/jumpstart/artifacts.py

@@ -314,7 +314,7 @@ def _retrieve_default_hyperparameters(
 
     assert region is not None
 
-    model_specs = jumpstart_accessors.JumpStartModelsCache.get_model_specs(
+    model_specs = jumpstart_accessors.JumpStartModelsAccessor.get_model_specs(
         region=region, model_id=model_id, version=model_version
     )
 
@@ -351,7 +351,7 @@ def _retrieve_default_environment_variables(
     if region is None:
         region = JUMPSTART_DEFAULT_REGION_NAME
 
-    model_specs = jumpstart_accessors.JumpStartModelsCache.get_model_specs(
+    model_specs = jumpstart_accessors.JumpStartModelsAccessor.get_model_specs(
         region=region, model_id=model_id, version=model_version
     )
 

tests/unit/sagemaker/environment_variables/jumpstart/test_default.py

@@ -21,7 +21,7 @@
 from tests.unit.sagemaker.jumpstart.utils import get_spec_from_base_spec
 
 
-@patch("sagemaker.jumpstart.accessors.JumpStartModelsCache.get_model_specs")
+@patch("sagemaker.jumpstart.accessors.JumpStartModelsAccessor.get_model_specs")
 def test_jumpstart_default_environment_variables(patched_get_model_specs):
 
     patched_get_model_specs.side_effect = get_spec_from_base_spec

tests/unit/sagemaker/hyperparameters/jumpstart/test_default.py

@@ -21,7 +21,7 @@
 from tests.unit.sagemaker.jumpstart.utils import get_spec_from_base_spec
 
 
-@patch("sagemaker.jumpstart.accessors.JumpStartModelsCache.get_model_specs")
+@patch("sagemaker.jumpstart.accessors.JumpStartModelsAccessor.get_model_specs")
 def test_jumpstart_default_hyperparameters(patched_get_model_specs):
 
     patched_get_model_specs.side_effect = get_spec_from_base_spec