Skip to content

Commit c5747ad

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 3d36503 commit c5747ad

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
@@ -338,12 +338,12 @@ def _parse_context_length(value) -> int:
338338

339339

340340
def _get_fine_tuning_options_and_model_arn(
341-
model_name: str,
342-
customization_technique: str,
343-
training_type,
344-
sagemaker_session,
345-
sequence_length=None,
346-
hub_name: str = "SageMakerPublicHub"
341+
model_name: str,
342+
customization_technique: str,
343+
training_type,
344+
sagemaker_session,
345+
sequence_length=None,
346+
hub_name: Optional[str] = None
347347
) -> tuple:
348348
"""Get fine-tuning options and model ARN for given customization technique.
349349

0 commit comments

Comments
 (0)