Skip to content

Commit e0d7443

Browse files
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. ```
1 parent 68885dc commit e0d7443

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)