Skip to content

Commit 967b71b

Browse files
committed
fix: fix instance_type assignment logic
1 parent ed43b07 commit 967b71b

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
@@ -322,7 +322,7 @@ def get_deploy_kwargs(
322322
model_id=model_id,
323323
model_from_estimator=True,
324324
model_version=model_version,
325-
instance_type=model_deploy_kwargs.instance_type if training_instance_type is None else None,
325+
instance_type=model_deploy_kwargs.instance_type or training_instance_type,
326326
region=region,
327327
image_uri=image_uri,
328328
source_dir=source_dir,

0 commit comments

Comments
 (0)