Skip to content

Commit

Permalink
Bump dask and pyarrow version in integration test (#979)
Browse files Browse the repository at this point in the history
  • Loading branch information
edknv authored May 17, 2023
1 parent 2ffd039 commit dbd2818
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions ci/test_integration.sh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#
# Copyright (c) 2022, NVIDIA CORPORATION.
# Copyright (c) 2023, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -24,11 +24,7 @@ devices=$2

# Run only for Merlin Tensorflow Container
if [ "$container" == "merlin-tensorflow" ]; then
# feast will install latest pyarrow version (currently 10.0.1)
# 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==9.0.0
pip install dask==2022.11.1 distributed==2022.11.1 protobuf==3.20.3
pip install 'feast<0.20'
pip install dask==2023.1.1 distributed==2023.1.1 protobuf==3.20.3
CUDA_VISIBLE_DEVICES="$devices" pytest -rxs tests/integration
fi

0 comments on commit dbd2818

Please sign in to comment.