From 1f565313d2768f367cab757eb2ee246a97df3198 Mon Sep 17 00:00:00 2001 From: jesterhazy Date: Mon, 24 Jun 2019 12:35:38 -0700 Subject: [PATCH 1/2] fix: add pytest.mark.local_mode annotation to broken tests --- tests/integ/test_tfs.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/integ/test_tfs.py b/tests/integ/test_tfs.py index a53ed52c2a..8a8c7bf8f7 100644 --- a/tests/integ/test_tfs.py +++ b/tests/integ/test_tfs.py @@ -69,6 +69,7 @@ def tar_dir(directory, tmpdir): return target +@pytest.mark.local_mode @pytest.fixture def tfs_predictor_with_model_and_entry_point_same_tar( sagemaker_local_session, tf_full_version, tmpdir @@ -93,6 +94,7 @@ def tfs_predictor_with_model_and_entry_point_same_tar( predictor.delete_endpoint() +@pytest.mark.local_mode @pytest.fixture(scope="module") def tfs_predictor_with_model_and_entry_point_and_dependencies( sagemaker_local_session, tf_full_version From f28b2b0acbb9d0894bfacdbaab898019db9dc0c3 Mon Sep 17 00:00:00 2001 From: jesterhazy Date: Mon, 24 Jun 2019 14:23:40 -0700 Subject: [PATCH 2/2] move local_mode mark to tests --- tests/integ/test_tfs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integ/test_tfs.py b/tests/integ/test_tfs.py index 8a8c7bf8f7..8d8f8083ef 100644 --- a/tests/integ/test_tfs.py +++ b/tests/integ/test_tfs.py @@ -69,7 +69,6 @@ def tar_dir(directory, tmpdir): return target -@pytest.mark.local_mode @pytest.fixture def tfs_predictor_with_model_and_entry_point_same_tar( sagemaker_local_session, tf_full_version, tmpdir @@ -94,7 +93,6 @@ def tfs_predictor_with_model_and_entry_point_same_tar( predictor.delete_endpoint() -@pytest.mark.local_mode @pytest.fixture(scope="module") def tfs_predictor_with_model_and_entry_point_and_dependencies( sagemaker_local_session, tf_full_version @@ -173,6 +171,7 @@ def test_predict_with_accelerator(tfs_predictor_with_accelerator): assert expected_result == result +@pytest.mark.local_mode def test_predict_with_entry_point(tfs_predictor_with_model_and_entry_point_same_tar): input_data = {"instances": [1.0, 2.0, 5.0]} expected_result = {"predictions": [4.0, 4.5, 6.0]} @@ -181,6 +180,7 @@ def test_predict_with_entry_point(tfs_predictor_with_model_and_entry_point_same_ assert expected_result == result +@pytest.mark.local_mode def test_predict_with_model_and_entry_point_and_dependencies_separated( tfs_predictor_with_model_and_entry_point_and_dependencies ):