We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a659683 commit 01d2e52Copy full SHA for 01d2e52
tests/integ/test_tfs.py
@@ -171,6 +171,7 @@ def test_predict_with_accelerator(tfs_predictor_with_accelerator):
171
assert expected_result == result
172
173
174
+@pytest.mark.local_mode
175
def test_predict_with_entry_point(tfs_predictor_with_model_and_entry_point_same_tar):
176
input_data = {"instances": [1.0, 2.0, 5.0]}
177
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_
179
180
181
182
183
184
def test_predict_with_model_and_entry_point_and_dependencies_separated(
185
tfs_predictor_with_model_and_entry_point_and_dependencies
186
):
0 commit comments