Skip to content

Commit e5de0c2

Browse files
Yadan-WeiYadan Weiarjkesh
authored
test upload files to s3 bucket (#4353)
* test upload files to s3 bucket * print out file path * Re-enable TF 2.16 graviton images (#4354) * reformat and change bucket name * trigger tests * rerun * update tf model path with test/ * change directory of BATCH_CSV * enable all sm tests * restore s3 bucket name and toml file * remove directory print --------- Co-authored-by: Yadan Wei <[email protected]> Co-authored-by: arjkesh <[email protected]>
1 parent 372659c commit e5de0c2

File tree

4 files changed

+6
-5
lines changed

4 files changed

+6
-5
lines changed

test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/test_tfs.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,9 @@
1818
from ...... import invoke_sm_endpoint_helper_function
1919

2020

21-
MODEL_WITH_REQUIREMENTS_PATH = "data/tfs-model_greater_than_equal_to_tf26.tar.gz"
22-
MODEL_WITH_LIB_PATH = "data/python-with-lib.tar.gz"
23-
TFS_MODEL_PATH = "data/tfs-model.tar.gz"
21+
MODEL_WITH_REQUIREMENTS_PATH = "test/data/tfs-model_greater_than_equal_to_tf26.tar.gz"
22+
MODEL_WITH_LIB_PATH = "test/data/python-with-lib.tar.gz"
23+
TFS_MODEL_PATH = "test/data/tfs-model.tar.gz"
2424
TFS_NEURONX_MODEL_PATH = "data/tfs-neuronx-model.tar.gz"
2525
TFS_NEURON_MODEL_PATH = "data/tfs-neuron-model.tar.gz"
2626
MME1_MODEL_PATHS = [

test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/util.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,12 @@
2020
import time
2121
import sys
2222

23+
from test.test_utils import LOGGER
24+
2325
logger = logging.getLogger(__name__)
2426
logger.setLevel(logging.DEBUG)
2527
logger.addHandler(logging.StreamHandler(sys.stdout))
26-
BATCH_CSV = os.path.join("data", "batch.csv")
28+
BATCH_CSV = os.path.join("test", "data", "batch.csv")
2729

2830

2931
def _botocore_resolver():
@@ -174,7 +176,6 @@ def find_or_put_mme_model_data(region, boto_session, mme_folder_name, path_list)
174176
except botocore.exceptions.ClientError as e:
175177
if e.response["Error"]["Code"] != "404":
176178
raise
177-
178179
# file doesn't exist - upload it
179180
s3.upload_file(local_path, bucket, key)
180181

0 commit comments

Comments
 (0)