Skip to content

Commit 5449eb5

Browse files
author
Malav Shastri
committed
fix codestyle issues
1 parent 10dba2c commit 5449eb5

File tree

4 files changed

+13
-5
lines changed

4 files changed

+13
-5
lines changed

src/sagemaker/jumpstart/accessors.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,8 @@ def get_model_specs(
301301

302302
except Exception as ex:
303303
logging.info(
304-
"Recieved exeption while calling APIs for ContentType Model, retrying with ContentType ModelReference: " + str(ex)
304+
"Recieved exeption while calling APIs for ContentType Model, retrying with ContentType ModelReference: "
305+
+ str(ex)
305306
)
306307
hub_model_arn = construct_hub_model_arn_from_inputs(
307308
hub_arn=hub_arn, model_name=model_id, version=version

src/sagemaker/jumpstart/factory/estimator.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,10 @@
2929
_retrieve_model_package_model_artifact_s3_uri,
3030
)
3131
from sagemaker.jumpstart.artifacts.resource_names import _retrieve_resource_name_base
32-
from sagemaker.jumpstart.hub.utils import construct_hub_model_arn_from_inputs, construct_hub_model_reference_arn_from_inputs
32+
from sagemaker.jumpstart.hub.utils import (
33+
construct_hub_model_arn_from_inputs,
34+
construct_hub_model_reference_arn_from_inputs,
35+
)
3336
from sagemaker.session import Session
3437
from sagemaker.async_inference.async_inference_config import AsyncInferenceConfig
3538
from sagemaker.base_deserializers import BaseDeserializer
@@ -549,11 +552,12 @@ def _add_image_uri_to_kwargs(kwargs: JumpStartEstimatorInitKwargs) -> JumpStartE
549552

550553
return kwargs
551554

555+
552556
def _add_model_reference_arn_to_kwargs(
553557
kwargs: JumpStartEstimatorInitKwargs,
554558
) -> JumpStartEstimatorInitKwargs:
555559
"""Sets Model Reference ARN if the hub content type is Model Reference, returns full kwargs."""
556-
560+
557561
hub_content_type = verify_model_region_and_return_specs(
558562
model_id=kwargs.model_id,
559563
version=kwargs.model_version,

src/sagemaker/jumpstart/factory/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,7 @@ def _add_model_reference_arn_to_kwargs(
268268
kwargs: JumpStartModelInitKwargs,
269269
) -> JumpStartModelInitKwargs:
270270
"""Sets Model Reference ARN if the hub content type is Model Reference, returns full kwargs."""
271-
271+
272272
hub_content_type = verify_model_region_and_return_specs(
273273
model_id=kwargs.model_id,
274274
version=kwargs.model_version,

src/sagemaker/jumpstart/hub/hub.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,10 @@ def describe_model(
288288
)
289289

290290
except Exception as ex:
291-
logging.info("Recieved exeption while calling APIs for ContentType Model, retrying with ContentType ModelReference: " + str(ex))
291+
logging.info(
292+
"Recieved exeption while calling APIs for ContentType Model, retrying with ContentType ModelReference: "
293+
+ str(ex)
294+
)
292295
model_version = get_hub_model_version(
293296
hub_model_name=model_name,
294297
hub_model_type=HubContentType.MODEL_REFERENCE.value,

0 commit comments

Comments
 (0)