Skip to content
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions tests/integ/test_tfs.py
Original file line number Diff line number Diff line change
Expand Up @@ -171,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]}
Expand All @@ -179,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
):
Expand Down