Skip to content

Commit 01d2e52

Browse files
authored
fix: add pytest.mark.local_mode annotation to broken tests (#876)
* fix: add pytest.mark.local_mode annotation to tests
1 parent a659683 commit 01d2e52

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/integ/test_tfs.py

+2
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,7 @@ def test_predict_with_accelerator(tfs_predictor_with_accelerator):
171171
assert expected_result == result
172172

173173

174+
@pytest.mark.local_mode
174175
def test_predict_with_entry_point(tfs_predictor_with_model_and_entry_point_same_tar):
175176
input_data = {"instances": [1.0, 2.0, 5.0]}
176177
expected_result = {"predictions": [4.0, 4.5, 6.0]}
@@ -179,6 +180,7 @@ def test_predict_with_entry_point(tfs_predictor_with_model_and_entry_point_same_
179180
assert expected_result == result
180181

181182

183+
@pytest.mark.local_mode
182184
def test_predict_with_model_and_entry_point_and_dependencies_separated(
183185
tfs_predictor_with_model_and_entry_point_and_dependencies
184186
):

0 commit comments

Comments
 (0)