fix: renaming

Author: chrstfu

src/sagemaker/jumpstart/curated_hub/curated_hub.py

@@ -50,7 +50,7 @@
     create_s3_object_reference_from_uri,
     tag_hub_content,
     get_jumpstart_model_and_version,
-    find_all_tags_for_jumpstart_model
+    find_jumpstart_tags_for_model
 )
 from sagemaker.jumpstart.curated_hub.types import (
     HubContentDocument_v2,
@@ -425,7 +425,7 @@ def scan_and_tag_models(self) -> None:
         models_in_hub: List[Dict[str, Any]] = self._get_jumpstart_models_in_hub()
         tags_added: List[Dict[str, List[Dict[str, str]]]] = []
         for model in models_in_hub:
-            tags_to_add: List[Tag] = find_all_tags_for_jumpstart_model(
+            tags_to_add: List[Tag] = find_jumpstart_tags_for_model(
                 hub_name=self.hub_name,
                 hub_content_name=model["HubContentName"],
                 region=self.region,

src/sagemaker/jumpstart/curated_hub/utils.py

@@ -195,7 +195,7 @@ def tag_hub_content(hub_content_arn: str, tags: List[Tag], session: Session) ->
 
     return responses
 
-def find_all_tags_for_jumpstart_model(hub_name: str, hub_content_name: str, region: str, session: Session) -> List[Tag]:
+def find_jumpstart_tags_for_model(hub_name: str, hub_content_name: str, region: str, session: Session) -> List[Tag]:
     list_versions_response = session.list_hub_content_versions(
         hub_name=hub_name,
         hub_content_type='Model',
@@ -208,7 +208,7 @@ def find_all_tags_for_jumpstart_model(hub_name: str, hub_content_name: str, regi
         jumpstart_model = get_jumpstart_model_and_version(hub_content_version_summary)
         if jumpstart_model["model_id"] is None or jumpstart_model["version"] is None:
             continue
-        tag_names_to_add: List[CuratedHubTagName] = find_tags_for_jumpstart_model_version(
+        tag_names_to_add: List[CuratedHubTagName] = find_jumpstart_tags_for_model_version(
             model_id=jumpstart_model["model_id"],
             version=jumpstart_model["version"],
             region=region,
@@ -231,7 +231,7 @@ def find_all_tags_for_jumpstart_model(hub_name: str, hub_content_name: str, regi
 
 
 
-def find_tags_for_jumpstart_model_version(model_id: str, version: str, region: str, session: Session) -> List[CuratedHubTagName]:
+def find_jumpstart_tags_for_model_version(model_id: str, version: str, region: str, session: Session) -> List[CuratedHubTagName]:
     tags_to_add: List[CuratedHubTagName] = []
     specs = utils.verify_model_region_and_return_specs(
         model_id=model_id, 

tests/unit/sagemaker/jumpstart/curated_hub/test_utils.py

@@ -184,7 +184,7 @@ def test_find_tags_for_jumpstart_model_version(mock_spec_util):
     mock_specs.training_vulnerable = True
     mock_spec_util.return_value = mock_specs
 
-    tags = utils.find_tags_for_jumpstart_model_version(
+    tags = utils.find_jumpstart_tags_for_model_version(
         model_id="test",
         version="test",
         region="test",
@@ -212,7 +212,7 @@ def test_find_tags_for_jumpstart_model_version_some_false(mock_spec_util):
     mock_specs.training_vulnerable = False
     mock_spec_util.return_value = mock_specs
 
-    tags = utils.find_tags_for_jumpstart_model_version(
+    tags = utils.find_jumpstart_tags_for_model_version(
         model_id="test",
         version="test",
         region="test",
@@ -263,7 +263,7 @@ def test_find_all_tags_for_jumpstart_model(mock_spec_util):
     mock_specs.training_vulnerable = True
     mock_spec_util.return_value = mock_specs
 
-    tags = utils.find_all_tags_for_jumpstart_model(
+    tags = utils.find_jumpstart_tags_for_model(
         hub_name="test",
         hub_content_name="test",
         region="test",