From 48e0c702afc73d8cbb8067c06074b425cb26725f Mon Sep 17 00:00:00 2001 From: edknv <109497216+edknv@users.noreply.github.com> Date: Mon, 8 Jan 2024 14:05:24 -0800 Subject: [PATCH] move merlin dependecy install in horovod-cpu tests (#1235) * move merlin dependecy install in horovod-cpu tests * trigger horovod test --- tox.ini | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tox.ini b/tox.ini index a6c20fa764..af00c4eae5 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}