Skip to content

Commit 63dab51

Browse files
author
shaernev
committed
fix: include all known fields directly in create_sagemaker_model function signature
1 parent 37a4038 commit 63dab51

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/sagemaker/model.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1377,6 +1377,7 @@ def deploy(
13771377
managed_instance_scaling: Optional[str] = None,
13781378
inference_component_name=None,
13791379
routing_config: Optional[Dict[str, Any]] = None,
1380+
model_reference_arn: Optional[str] = None,
13801381
**kwargs,
13811382
):
13821383
"""Deploy this ``Model`` to an ``Endpoint`` and optionally return a ``Predictor``.
@@ -1483,6 +1484,8 @@ def deploy(
14831484
{
14841485
"RoutingStrategy": sagemaker.enums.RoutingStrategy.RANDOM
14851486
}
1487+
model_reference_arn (Optional [str]): Hub Content Arn of a Model Reference type
1488+
content (default: None).
14861489
Raises:
14871490
ValueError: If arguments combination check failed in these circumstances:
14881491
- If no role is specified or
@@ -1696,7 +1699,9 @@ def deploy(
16961699
instance_type=instance_type,
16971700
accelerator_type=accelerator_type,
16981701
tags=tags,
1699-
serverless_inference_config=serverless_inference_config
1702+
serverless_inference_config=serverless_inference_config,
1703+
accept_eula=accept_eula,
1704+
model_reference_arn=model_reference_arn
17001705
)
17011706
serverless_inference_config_dict = (
17021707
serverless_inference_config._to_request_dict() if is_serverless else None

0 commit comments

Comments
 (0)