chore: improve function name

Author: evakravi

src/sagemaker/instance_types.py

@@ -73,7 +73,7 @@ def retrieve_default(
     )
 
 
-def retrieve_supported(
+def retrieve(
     region=None,
     model_id=None,
     model_version=None,
@@ -111,7 +111,7 @@ def retrieve_supported(
     if scope is None:
         raise ValueError("Must specify scope for instance types.")
 
-    return artifacts._retrieve_supported_instance_types(
+    return artifacts._retrieve_instance_types(
         model_id,
         model_version,
         scope,

src/sagemaker/jumpstart/artifacts.py

@@ -463,7 +463,7 @@ def _retrieve_default_instance_type(
     return default_instance_type
 
 
-def _retrieve_supported_instance_types(
+def _retrieve_instance_types(
     model_id: str,
     model_version: str,
     scope: str,

tests/unit/sagemaker/instance_types/jumpstart/test_instance_types.py

@@ -51,7 +51,7 @@ def test_jumpstart_instance_types(patched_get_model_specs):
 
     patched_get_model_specs.reset_mock()
 
-    default_training_instance_types = instance_types.retrieve_supported(
+    default_training_instance_types = instance_types.retrieve(
         region=region, model_id=model_id, model_version=model_version, scope="training"
     )
     assert default_training_instance_types == [
@@ -68,7 +68,7 @@ def test_jumpstart_instance_types(patched_get_model_specs):
 
     patched_get_model_specs.reset_mock()
 
-    default_inference_instance_types = instance_types.retrieve_supported(
+    default_inference_instance_types = instance_types.retrieve(
         region=region, model_id=model_id, model_version=model_version, scope="inference"
     )
     assert default_inference_instance_types == [
@@ -95,7 +95,7 @@ def test_jumpstart_instance_types(patched_get_model_specs):
         )
 
     with pytest.raises(ValueError):
-        instance_types.retrieve_supported(
+        instance_types.retrieve(
             region=region,
             model_id=model_id,
             model_version="*",
@@ -118,20 +118,20 @@ def test_jumpstart_instance_types(patched_get_model_specs):
         instance_types.retrieve_default(model_id=model_id, scope="training")
 
     with pytest.raises(KeyError):
-        instance_types.retrieve_supported(
+        instance_types.retrieve(
             region=region, model_id="blah", model_version="*", scope="inference"
         )
 
     with pytest.raises(ValueError):
-        instance_types.retrieve_supported(
+        instance_types.retrieve(
             region="mars-south-1", model_id=model_id, model_version="*", scope="training"
         )
 
     with pytest.raises(ValueError):
-        instance_types.retrieve_supported(model_version="*", scope="inference")
+        instance_types.retrieve(model_version="*", scope="inference")
 
     with pytest.raises(ValueError):
-        instance_types.retrieve_supported(model_id=model_id, scope="training")
+        instance_types.retrieve(model_id=model_id, scope="training")
 
 
 @patch("sagemaker.jumpstart.accessors.JumpStartModelsAccessor.get_model_specs")
@@ -152,11 +152,11 @@ def test_jumpstart_no_supported_instance_types(patched_get_model_specs):
         )
 
     with pytest.raises(ValueError):
-        instance_types.retrieve_supported(
+        instance_types.retrieve(
             region=region, model_id=model_id, model_version=model_version, scope="training"
         )
 
     with pytest.raises(ValueError):
-        instance_types.retrieve_supported(
+        instance_types.retrieve(
             region=region, model_id=model_id, model_version=model_version, scope="inference"
         )