Skip to content

Commit 73bdd08

Browse files
authored
add sleep for model deployment (#5260)
1 parent 9bfe85a commit 73bdd08

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/integ/test_multidatamodel.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414

1515
import base64
1616
import os
17+
import time
1718
import requests
1819

1920
import docker
@@ -138,6 +139,7 @@ def test_multi_data_model_deploy_pretrained_models(
138139
multi_data_model.add_model(pretrained_model_data_local_path, PRETRAINED_MODEL_PATH_1)
139140
# Deploy model to an endpoint
140141
multi_data_model.deploy(1, cpu_instance_type, endpoint_name=endpoint_name)
142+
time.sleep(30)
141143
# Add models after deploy
142144
multi_data_model.add_model(pretrained_model_data_local_path, PRETRAINED_MODEL_PATH_2)
143145

@@ -266,6 +268,7 @@ def test_multi_data_model_deploy_trained_model_from_framework_estimator(
266268
multi_data_model.add_model(mxnet_model_1.model_data, PRETRAINED_MODEL_PATH_1)
267269
# Deploy model to an endpoint
268270
multi_data_model.deploy(1, cpu_instance_type, endpoint_name=endpoint_name)
271+
time.sleep(30)
269272

270273
# Train another model
271274
mxnet_model_2 = _mxnet_training_job(
@@ -373,6 +376,7 @@ def test_multi_data_model_deploy_train_model_from_amazon_first_party_estimator(
373376
multi_data_model.add_model(rcf_model_v1.model_data, PRETRAINED_MODEL_PATH_1)
374377
# Deploy model to an endpoint
375378
multi_data_model.deploy(1, cpu_instance_type, endpoint_name=endpoint_name)
379+
time.sleep(30)
376380
# Train another model
377381
rcf_model_v2 = __rcf_training_job(
378382
sagemaker_session, container_image, cpu_instance_type, 70, 20
@@ -470,6 +474,7 @@ def test_multi_data_model_deploy_pretrained_models_update_endpoint(
470474
multi_data_model.add_model(pretrained_model_data_local_path, PRETRAINED_MODEL_PATH_1)
471475
# Deploy model to an endpoint
472476
multi_data_model.deploy(1, cpu_instance_type, endpoint_name=endpoint_name)
477+
time.sleep(30)
473478
# Add model after deploy
474479
multi_data_model.add_model(pretrained_model_data_local_path, PRETRAINED_MODEL_PATH_2)
475480

0 commit comments

Comments
 (0)