Skip to content

Commit ed93b9e

Browse files
committed
fix some formatting
1 parent 3b84dfe commit ed93b9e

File tree

3 files changed

+2
-2
lines changed

3 files changed

+2
-2
lines changed

src/sagemaker/estimator.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2511,7 +2511,6 @@ def start_new(cls, estimator, inputs, experiment_config):
25112511
train_args = cls._get_train_args(estimator, inputs, experiment_config)
25122512

25132513
logger.debug("Train args after processing defaults: %s", train_args)
2514-
25152514
estimator.sagemaker_session.train(**train_args)
25162515

25172516
return cls(estimator.sagemaker_session, estimator._current_job_name)

src/sagemaker/jumpstart/factory/estimator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -313,6 +313,7 @@ def _add_hub_access_config_to_kwargs_inputs(
313313
kwargs: JumpStartEstimatorFitKwargs, hub_access_config=None
314314
):
315315
"""Adds HubAccessConfig to kwargs inputs"""
316+
316317
dataset_uri = kwargs.specs.default_training_dataset_uri
317318
if isinstance(kwargs.inputs, str):
318319
if dataset_uri is not None and dataset_uri == kwargs.inputs:

src/sagemaker/jumpstart/types.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1940,7 +1940,7 @@ def use_inference_script_uri(self) -> bool:
19401940

19411941
def use_training_model_artifact(self) -> bool:
19421942
"""Returns True if the model should use a model uri when kicking off training job."""
1943-
# gated model never uses training model artifact
1943+
# gated model never use training model artifact
19441944
if self.gated_bucket:
19451945
return False
19461946

0 commit comments

Comments
 (0)