Skip to content

Commit cd04529

Browse files
committed
fix: pass in inference_ami_version to model_based endpoint type
1 parent 13ad978 commit cd04529

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/sagemaker/model.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1492,6 +1492,9 @@ def deploy(
14921492
}
14931493
model_reference_arn (Optional [str]): Hub Content Arn of a Model Reference type
14941494
content (default: None).
1495+
inference_ami_version (Optional [str]): Specifies an option from a collection of preconfigured
1496+
Amazon Machine Image (AMI) images. For a full list of options, see:
1497+
https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ProductionVariant.html
14951498
Raises:
14961499
ValueError: If arguments combination check failed in these circumstances:
14971500
- If no role is specified or
@@ -1743,6 +1746,7 @@ def deploy(
17431746
model_data_download_timeout=model_data_download_timeout,
17441747
container_startup_health_check_timeout=container_startup_health_check_timeout,
17451748
routing_config=routing_config,
1749+
inference_ami_version=inference_ami_version,
17461750
)
17471751
if endpoint_name:
17481752
self.endpoint_name = endpoint_name

0 commit comments

Comments
 (0)