Skip to content

Commit 5392504

Browse files
author
Malav Shastri
committed
fix codestyle issues
1 parent 964de22 commit 5392504

File tree

2 files changed

+9
-8
lines changed

2 files changed

+9
-8
lines changed

src/sagemaker/jumpstart/accessors.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -300,7 +300,9 @@ def get_model_specs(
300300
return model_specs
301301

302302
except Exception as ex:
303-
logging.info("Recieved expection while calling APIs for ContentType Model: " + str(ex))
303+
logging.info(
304+
"Recieved expection while calling APIs for ContentType Model: " + str(ex)
305+
)
304306
hub_model_arn = construct_hub_model_arn_from_inputs(
305307
hub_arn=hub_arn, model_name=model_id, version=version
306308
)

src/sagemaker/jumpstart/factory/model.py

+6-7
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,10 @@
3434
JUMPSTART_LOGGER,
3535
)
3636
from sagemaker.model_card.model_card import ModelCard, ModelPackageModelCard
37-
from sagemaker.jumpstart.hub.utils import construct_hub_model_arn_from_inputs, construct_hub_model_reference_arn_from_inputs
37+
from sagemaker.jumpstart.hub.utils import (
38+
construct_hub_model_arn_from_inputs,
39+
construct_hub_model_reference_arn_from_inputs,
40+
)
3841
from sagemaker.model_metrics import ModelMetrics
3942
from sagemaker.metadata_properties import MetadataProperties
4043
from sagemaker.drift_check_baselines import DriftCheckBaselines
@@ -552,15 +555,11 @@ def _add_tags_to_kwargs(kwargs: JumpStartModelDeployKwargs) -> Dict[str, Any]:
552555
if kwargs.hub_arn:
553556
if kwargs.model_reference_arn:
554557
hub_content_arn = construct_hub_model_reference_arn_from_inputs(
555-
kwargs.hub_arn,
556-
kwargs.model_id,
557-
kwargs.model_version
558+
kwargs.hub_arn, kwargs.model_id, kwargs.model_version
558559
)
559560
else:
560561
hub_content_arn = construct_hub_model_arn_from_inputs(
561-
kwargs.hub_arn,
562-
kwargs.model_id,
563-
kwargs.model_version
562+
kwargs.hub_arn, kwargs.model_id, kwargs.model_version
564563
)
565564
kwargs.tags = add_hub_content_arn_tags(kwargs.tags, hub_content_arn=hub_content_arn)
566565

0 commit comments

Comments
 (0)