File tree 1 file changed +3
-3
lines changed
src/sagemaker/jumpstart/factory 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -169,7 +169,7 @@ def _add_vulnerable_and_deprecated_status_to_kwargs(
169
169
170
170
171
171
def _add_instance_type_to_kwargs (
172
- kwargs : JumpStartModelInitKwargs , disable_logging : bool = False
172
+ kwargs : JumpStartModelInitKwargs , disable_instance_type_logging : bool = False
173
173
) -> JumpStartModelInitKwargs :
174
174
"""Sets instance type based on default or override, returns full kwargs."""
175
175
@@ -186,7 +186,7 @@ def _add_instance_type_to_kwargs(
186
186
training_instance_type = kwargs .training_instance_type ,
187
187
)
188
188
189
- if not disable_logging and orig_instance_type is None :
189
+ if not disable_instance_type_logging and orig_instance_type is None :
190
190
JUMPSTART_LOGGER .info (
191
191
"No instance type selected for inference hosting endpoint. Defaulting to %s." ,
192
192
kwargs .instance_type ,
@@ -687,7 +687,7 @@ def get_init_kwargs(
687
687
model_init_kwargs = _add_model_name_to_kwargs (kwargs = model_init_kwargs )
688
688
689
689
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
691
691
)
692
692
693
693
model_init_kwargs = _add_image_uri_to_kwargs (kwargs = model_init_kwargs )
You can’t perform that action at this time.
0 commit comments