|
14 | 14 |
|
15 | 15 | import base64 |
16 | 16 | import os |
| 17 | +import time |
17 | 18 | import requests |
18 | 19 |
|
19 | 20 | import docker |
@@ -138,6 +139,7 @@ def test_multi_data_model_deploy_pretrained_models( |
138 | 139 | multi_data_model.add_model(pretrained_model_data_local_path, PRETRAINED_MODEL_PATH_1) |
139 | 140 | # Deploy model to an endpoint |
140 | 141 | multi_data_model.deploy(1, cpu_instance_type, endpoint_name=endpoint_name) |
| 142 | + time.sleep(30) |
141 | 143 | # Add models after deploy |
142 | 144 | multi_data_model.add_model(pretrained_model_data_local_path, PRETRAINED_MODEL_PATH_2) |
143 | 145 |
|
@@ -266,6 +268,7 @@ def test_multi_data_model_deploy_trained_model_from_framework_estimator( |
266 | 268 | multi_data_model.add_model(mxnet_model_1.model_data, PRETRAINED_MODEL_PATH_1) |
267 | 269 | # Deploy model to an endpoint |
268 | 270 | multi_data_model.deploy(1, cpu_instance_type, endpoint_name=endpoint_name) |
| 271 | + time.sleep(30) |
269 | 272 |
|
270 | 273 | # Train another model |
271 | 274 | mxnet_model_2 = _mxnet_training_job( |
@@ -373,6 +376,7 @@ def test_multi_data_model_deploy_train_model_from_amazon_first_party_estimator( |
373 | 376 | multi_data_model.add_model(rcf_model_v1.model_data, PRETRAINED_MODEL_PATH_1) |
374 | 377 | # Deploy model to an endpoint |
375 | 378 | multi_data_model.deploy(1, cpu_instance_type, endpoint_name=endpoint_name) |
| 379 | + time.sleep(30) |
376 | 380 | # Train another model |
377 | 381 | rcf_model_v2 = __rcf_training_job( |
378 | 382 | sagemaker_session, container_image, cpu_instance_type, 70, 20 |
@@ -470,6 +474,7 @@ def test_multi_data_model_deploy_pretrained_models_update_endpoint( |
470 | 474 | multi_data_model.add_model(pretrained_model_data_local_path, PRETRAINED_MODEL_PATH_1) |
471 | 475 | # Deploy model to an endpoint |
472 | 476 | multi_data_model.deploy(1, cpu_instance_type, endpoint_name=endpoint_name) |
| 477 | + time.sleep(30) |
473 | 478 | # Add model after deploy |
474 | 479 | multi_data_model.add_model(pretrained_model_data_local_path, PRETRAINED_MODEL_PATH_2) |
475 | 480 |
|
|
0 commit comments