Skip to content

Commit ce7a99b

Browse files
committed
linting
1 parent 0730ae1 commit ce7a99b

File tree

2 files changed

+10
-8
lines changed

2 files changed

+10
-8
lines changed

src/sagemaker/jumpstart/accessors.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -287,9 +287,9 @@ def get_model_specs(
287287
{**JumpStartModelsAccessor._cache_kwargs, **additional_kwargs}
288288
)
289289
JumpStartModelsAccessor._set_cache_and_region(region, cache_kwargs)
290-
290+
291291
# Users only input model id, not contentType, so first try to describe with ModelReference, then with Model
292-
if hub_arn:
292+
if hub_arn:
293293
try:
294294
hub_model_arn = construct_hub_model_reference_arn_from_inputs(
295295
hub_arn=hub_arn, model_name=model_id, version=version

src/sagemaker/jumpstart/hub/hub.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -308,13 +308,15 @@ def describe_model(
308308
hub_model_version=model_version,
309309
)
310310

311-
hub_content_description: Dict[str, Any] = self._sagemaker_session.describe_hub_content(
312-
hub_name=hub_name,
313-
hub_content_name=model_name,
314-
hub_content_version=model_version,
315-
hub_content_type=HubContentType.MODEL.value,
311+
hub_content_description: Dict[str, Any] = (
312+
self._sagemaker_session.describe_hub_content(
313+
hub_name=hub_name,
314+
hub_content_name=model_name,
315+
hub_content_version=model_version,
316+
hub_content_type=HubContentType.MODEL.value,
317+
)
316318
)
317-
319+
318320
except Exception as ex:
319321
# Failed with both, throw a custom error message
320322
raise Exception(

0 commit comments

Comments
 (0)