Skip to content

Commit

Permalink
Update dependency versions in integration tests (#885)
Browse files Browse the repository at this point in the history
  • Loading branch information
edknv authored Mar 29, 2023
1 parent 2d377fb commit 6d0af4a
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
4 changes: 2 additions & 2 deletions ci/test_integration.sh
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ if [ "$container" == "merlin-tensorflow" ]; then
# this vesrion of pyarrow is incompatibile
# with the current version of cudf 22.12
# pinning the version of pyarrow here to match the cudf-supported version
pip install 'feast<0.20' pyarrow==8.0.0
pip install dask==2022.07.1 distributed==2022.07.1
pip install 'feast<0.20' pyarrow==9.0.0
pip install dask==2022.11.1 distributed==2022.11.1
CUDA_VISIBLE_DEVICES="$devices" pytest -rxs tests/integration
fi
Original file line number Diff line number Diff line change
Expand Up @@ -79,9 +79,9 @@ def test_func():
response = run_ensemble_on_tritonserver(
"/tmp/examples/poc_ensemble", ensemble.graph.input_schema, request, outputs, "executor_model"
)
response = [x.tolist()[0] for x in response["ordered_ids"]]
ordered_ids = [x.tolist() for x in response["ordered_ids"]]
shutil.rmtree("/tmp/examples/", ignore_errors=True)
"""
)
response = tb2.ref("response")
assert len(response) == top_k
ordered_ids = tb2.ref("ordered_ids")
assert len(ordered_ids[0]) == top_k

0 comments on commit 6d0af4a

Please sign in to comment.