From 6bb5f006bf4a08bbedacb9e609c2a77866e801c8 Mon Sep 17 00:00:00 2001 From: edknv Date: Sat, 6 Jan 2024 19:05:15 -0800 Subject: [PATCH] move merlin dependecy install in horovod-cpu tests --- tox.ini | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tox.ini b/tox.ini index e35a7dcc78..b29863e173 100644 --- a/tox.ini +++ b/tox.ini @@ -57,14 +57,13 @@ setenv = HOROVOD_WITH_TENSORFLOW=1 PATH={env:PATH}{:}{envdir}/env/bin LD_LIBRARY_PATH={env:LD_LIBRARY_PATH}{:}{envdir}/env/lib -deps = - git+https://github.com/NVIDIA-Merlin/core.git@{env:MERLIN_BRANCH} - git+https://github.com/NVIDIA-Merlin/dataloader.git@{env:MERLIN_BRANCH} - git+https://github.com/NVIDIA-Merlin/NVTabular.git@{env:MERLIN_BRANCH} commands = conda update --yes --name base --channel defaults conda conda env create --prefix {envdir}/env --file requirements/horovod-cpu-environment.yml --force - {envdir}/env/bin/python -m pip install 'horovod==0.27.0' --no-cache-dir + {envdir}/env/bin/python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/core.git@{env:MERLIN_BRANCH:main} + {envdir}/env/bin/python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/dataloader.git@{env:MERLIN_BRANCH:main} + {envdir}/env/bin/python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/NVTabular.git@{env:MERLIN_BRANCH:main} + {envdir}/env/bin/python -m pip install 'horovod==0.28.1' --no-cache-dir {envdir}/env/bin/horovodrun --check-build {envdir}/env/bin/horovodrun -np 2 sh examples/usecases/multi-gpu/hvd_wrapper.sh pytest -m "unit and horovod {env:PYTEST_MARKERS}" -rxs {posargs:tests}