add in reinstall for dask and distributed after feast install (NVIDIA…

…-Merlin#713)
nv-alaiacano · Nov 2, 2022 · a4f82eb · a4f82eb
1 parent 6b67c26
commit a4f82eb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ci/test_integration.sh b/ci/test_integration.sh
@@ -25,5 +25,6 @@ devices=$2
 # Run only for Merlin Tensorflow Container
 if [ "$container" == "merlin-tensorflow" ]; then
     pip install 'feast<0.20'
+    pip install dask==2022.07.1 distributed==2022.07.1
     CUDA_VISIBLE_DEVICES="$devices"  pytest -rxs tests/integration
 fi