Skip to content

Commit 4844aa1

Browse files
fix: Remove deprecated update_endpoint from deploy() args in TensorFlowModel (#3824)
1 parent c68f5b0 commit 4844aa1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/tensorflow/model.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -317,14 +317,14 @@ def deploy(
317317
kms_key=None,
318318
wait=True,
319319
data_capture_config=None,
320-
update_endpoint=None,
321320
async_inference_config=None,
322321
serverless_inference_config=None,
323322
volume_size=None,
324323
model_data_download_timeout=None,
325324
container_startup_health_check_timeout=None,
326325
inference_recommendation_id=None,
327326
explainer_config=None,
327+
**kwargs,
328328
):
329329
"""Deploy a Tensorflow ``Model`` to a SageMaker ``Endpoint``."""
330330

@@ -348,9 +348,9 @@ def deploy(
348348
volume_size=volume_size,
349349
model_data_download_timeout=model_data_download_timeout,
350350
container_startup_health_check_timeout=container_startup_health_check_timeout,
351-
update_endpoint=update_endpoint,
352351
inference_recommendation_id=inference_recommendation_id,
353352
explainer_config=explainer_config,
353+
**kwargs,
354354
)
355355

356356
def _eia_supported(self):

0 commit comments

Comments
 (0)