diff --git a/src/sagemaker/tensorflow/model.py b/src/sagemaker/tensorflow/model.py index 4b9c23621b..1f162cef60 100644 --- a/src/sagemaker/tensorflow/model.py +++ b/src/sagemaker/tensorflow/model.py @@ -317,7 +317,6 @@ def deploy( kms_key=None, wait=True, data_capture_config=None, - update_endpoint=None, async_inference_config=None, serverless_inference_config=None, volume_size=None, @@ -325,6 +324,7 @@ def deploy( container_startup_health_check_timeout=None, inference_recommendation_id=None, explainer_config=None, + **kwargs, ): """Deploy a Tensorflow ``Model`` to a SageMaker ``Endpoint``.""" @@ -348,9 +348,9 @@ def deploy( volume_size=volume_size, model_data_download_timeout=model_data_download_timeout, container_startup_health_check_timeout=container_startup_health_check_timeout, - update_endpoint=update_endpoint, inference_recommendation_id=inference_recommendation_id, explainer_config=explainer_config, + **kwargs, ) def _eia_supported(self):