Skip to content

Commit 8f886fe

Browse files
committed
fix: model type for estimator
1 parent df711e3 commit 8f886fe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/jumpstart/factory/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -917,7 +917,7 @@ def _add_config_name_to_kwargs(
917917
model_version=kwargs.model_version,
918918
sagemaker_session=kwargs.sagemaker_session,
919919
scope=JumpStartScriptScope.TRAINING,
920-
model_type=kwargs.model_type,
920+
model_type=kwargs.model_type[0],
921921
tolerate_deprecated_model=kwargs.tolerate_deprecated_model,
922922
tolerate_vulnerable_model=kwargs.tolerate_vulnerable_model,
923923
hub_arn=kwargs.hub_arn,

0 commit comments

Comments
 (0)