File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -69,7 +69,6 @@ def tar_dir(directory, tmpdir):
69
69
return target
70
70
71
71
72
- @pytest .mark .local_mode
73
72
@pytest .fixture
74
73
def tfs_predictor_with_model_and_entry_point_same_tar (
75
74
sagemaker_local_session , tf_full_version , tmpdir
@@ -94,7 +93,6 @@ def tfs_predictor_with_model_and_entry_point_same_tar(
94
93
predictor .delete_endpoint ()
95
94
96
95
97
- @pytest .mark .local_mode
98
96
@pytest .fixture (scope = "module" )
99
97
def tfs_predictor_with_model_and_entry_point_and_dependencies (
100
98
sagemaker_local_session , tf_full_version
@@ -173,6 +171,7 @@ def test_predict_with_accelerator(tfs_predictor_with_accelerator):
173
171
assert expected_result == result
174
172
175
173
174
+ @pytest .mark .local_mode
176
175
def test_predict_with_entry_point (tfs_predictor_with_model_and_entry_point_same_tar ):
177
176
input_data = {"instances" : [1.0 , 2.0 , 5.0 ]}
178
177
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_
181
180
assert expected_result == result
182
181
183
182
183
+ @pytest .mark .local_mode
184
184
def test_predict_with_model_and_entry_point_and_dependencies_separated (
185
185
tfs_predictor_with_model_and_entry_point_and_dependencies
186
186
):
You can’t perform that action at this time.
0 commit comments