Skip to content

Commit 5df5412

Browse files
committed
chore: improve improve kwarg name
1 parent de641fe commit 5df5412

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/sagemaker/jumpstart/factory/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ def _add_vulnerable_and_deprecated_status_to_kwargs(
169169

170170

171171
def _add_instance_type_to_kwargs(
172-
kwargs: JumpStartModelInitKwargs, disable_logging: bool = False
172+
kwargs: JumpStartModelInitKwargs, disable_instance_type_logging: bool = False
173173
) -> JumpStartModelInitKwargs:
174174
"""Sets instance type based on default or override, returns full kwargs."""
175175

@@ -186,7 +186,7 @@ def _add_instance_type_to_kwargs(
186186
training_instance_type=kwargs.training_instance_type,
187187
)
188188

189-
if not disable_logging and orig_instance_type is None:
189+
if not disable_instance_type_logging and orig_instance_type is None:
190190
JUMPSTART_LOGGER.info(
191191
"No instance type selected for inference hosting endpoint. Defaulting to %s.",
192192
kwargs.instance_type,
@@ -687,7 +687,7 @@ def get_init_kwargs(
687687
model_init_kwargs = _add_model_name_to_kwargs(kwargs=model_init_kwargs)
688688

689689
model_init_kwargs = _add_instance_type_to_kwargs(
690-
kwargs=model_init_kwargs, disable_logging=disable_instance_type_logging
690+
kwargs=model_init_kwargs, disable_instance_type_logging=disable_instance_type_logging
691691
)
692692

693693
model_init_kwargs = _add_image_uri_to_kwargs(kwargs=model_init_kwargs)

0 commit comments

Comments
 (0)