Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit f28b2b0

Browse files
committedJun 24, 2019
move local_mode mark to tests
1 parent 1f56531 commit f28b2b0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed
 

‎tests/integ/test_tfs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,6 @@ def tar_dir(directory, tmpdir):
6969
return target
7070

7171

72-
@pytest.mark.local_mode
7372
@pytest.fixture
7473
def tfs_predictor_with_model_and_entry_point_same_tar(
7574
sagemaker_local_session, tf_full_version, tmpdir
@@ -94,7 +93,6 @@ def tfs_predictor_with_model_and_entry_point_same_tar(
9493
predictor.delete_endpoint()
9594

9695

97-
@pytest.mark.local_mode
9896
@pytest.fixture(scope="module")
9997
def tfs_predictor_with_model_and_entry_point_and_dependencies(
10098
sagemaker_local_session, tf_full_version
@@ -173,6 +171,7 @@ def test_predict_with_accelerator(tfs_predictor_with_accelerator):
173171
assert expected_result == result
174172

175173

174+
@pytest.mark.local_mode
176175
def test_predict_with_entry_point(tfs_predictor_with_model_and_entry_point_same_tar):
177176
input_data = {"instances": [1.0, 2.0, 5.0]}
178177
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_
181180
assert expected_result == result
182181

183182

183+
@pytest.mark.local_mode
184184
def test_predict_with_model_and_entry_point_and_dependencies_separated(
185185
tfs_predictor_with_model_and_entry_point_and_dependencies
186186
):

0 commit comments

Comments
 (0)
Please sign in to comment.