Skip to content

Commit 7481009

Browse files
committed
fix: change hub_name default to None for consistency
Use Optional[str] = None instead of hardcoded "SageMakerPublicHub" default, letting get_sagemaker_hub_name() resolve it at runtime.
1 parent c90ff8e commit 7481009

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -427,12 +427,12 @@ def _parse_context_length(value) -> int:
427427

428428

429429
def _get_fine_tuning_options_and_model_arn(
430-
model_name: str,
431-
customization_technique: str,
432-
training_type,
433-
sagemaker_session,
434-
sequence_length=None,
435-
hub_name: str = "SageMakerPublicHub"
430+
model_name: str,
431+
customization_technique: str,
432+
training_type,
433+
sagemaker_session,
434+
sequence_length=None,
435+
hub_name: Optional[str] = None
436436
) -> tuple:
437437
"""Get fine-tuning options and model ARN for given customization technique.
438438

0 commit comments

Comments
 (0)