|
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