Skip to content

Commit 14a18dc

Browse files
karlhigleynv-alaiacanooliverholworthy
authored
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 <oholworthy@nvidia.com>
1 parent b7a2d6e commit 14a18dc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tox.ini

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ commands =
6060
; the GIT_COMMIT env is the current commit of the core repo
6161
; NOTE!!!! We must clean this up in the jenkins configuration with `rm -rf "nvtabular-$GIT_COMMIT"`
6262
git clone --depth 1 https://github.com/NVIDIA-Merlin/models.git models-{env:GIT_COMMIT}
63-
python -m pip install --upgrade ./models-{env:GIT_COMMIT}[dev,implicit,lightfm,tensorflow,torch]
63+
python -m pip install --upgrade ./models-{env:GIT_COMMIT}[dev,implicit,lightfm,tensorflow,torch,xgboost]
6464
python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/NVTabular.git
6565
python -m pip install tensorflow==2.9.2
6666
python -m pip install .

0 commit comments

Comments
 (0)