Skip to content

Commit ef7b9f4

Browse files
jinyoung-limbencrabtree
authored andcommitted
Model version default type to None
1 parent 020c294 commit ef7b9f4

File tree

2 files changed

+2
-5
lines changed

2 files changed

+2
-5
lines changed

src/sagemaker/jumpstart/curated_hub/curated_hub.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -174,10 +174,9 @@ def list_models(self, clear_cache: bool = True, **kwargs) -> List[Dict[str, Any]
174174
return self._list_hubs_cache
175175

176176
def describe_model(
177-
self, model_name: str, model_version: str = "*"
177+
self, model_name: str, model_version: Optional[str] = None
178178
) -> DescribeHubContentResponse:
179179
"""Returns descriptive information about the Hub Model"""
180-
181180
hub_content_description: Dict[str, Any] = self._sagemaker_session.describe_hub_content(
182181
hub_name=self.hub_name,
183182
hub_content_name=model_name,
@@ -187,7 +186,7 @@ def describe_model(
187186

188187
return DescribeHubContentResponse(hub_content_description)
189188

190-
def delete_model(self, model_name: str, model_version: str = "*") -> None:
189+
def delete_model(self, model_name: str, model_version: Optional[str] = None) -> None:
191190
"""Deletes a model from this CuratedHub."""
192191
return self._sagemaker_session.delete_hub_content(
193192
hub_content_name=model_name,

src/sagemaker/session.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6913,8 +6913,6 @@ def describe_hub_content(
69136913
}
69146914
if hub_content_version:
69156915
request["HubContentVersion"] = hub_content_version
6916-
print("@@@SESSION REQUEST@@@")
6917-
print(request)
69186916

69196917
return self.sagemaker_client.describe_hub_content(**request)
69206918

0 commit comments

Comments
 (0)