From c6a54bb65e69376dc67e011c2f312989401a655f Mon Sep 17 00:00:00 2001
From: Ivan Khvostishkov <87804596+ivan-khvostishkov@users.noreply.github.com>
Date: Fri, 28 Apr 2023 18:08:53 +0200
Subject: [PATCH] fix: Remove deprecated update_endpoint from args

Update_endpoint is a deprecated positional arg. Moved to kwargs to get rid of annoying warning:
```
DeprecationWarning: update_endpoint is a no-op in sagemaker>=2.
```
---
 src/sagemaker/tensorflow/model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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):