From 14a18dc0de5d5fd7737ecbadf9f6d7fa5d801b67 Mon Sep 17 00:00:00 2001 From: Karl Higley Date: Fri, 7 Oct 2022 14:36:24 -0400 Subject: [PATCH] Add XGBoost to `merlin-models` CPU tests (#131) Co-authored-by: Adam Laiacano <108741458+nv-alaiacano@users.noreply.github.com> Co-authored-by: Oliver Holworthy --- tox.ini | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tox.ini b/tox.ini index 7d3bcd73a..836e046a7 100644 --- a/tox.ini +++ b/tox.ini @@ -60,7 +60,7 @@ commands = ; the GIT_COMMIT env is the current commit of the core repo ; NOTE!!!! We must clean this up in the jenkins configuration with `rm -rf "nvtabular-$GIT_COMMIT"` git clone --depth 1 https://github.com/NVIDIA-Merlin/models.git models-{env:GIT_COMMIT} - python -m pip install --upgrade ./models-{env:GIT_COMMIT}[dev,implicit,lightfm,tensorflow,torch] + python -m pip install --upgrade ./models-{env:GIT_COMMIT}[dev,implicit,lightfm,tensorflow,torch,xgboost] python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/NVTabular.git python -m pip install tensorflow==2.9.2 python -m pip install .