Skip to content

Commit

Permalink
Add XGBoost to merlin-models CPU tests (#131)
Browse files Browse the repository at this point in the history
Co-authored-by: Adam Laiacano <[email protected]>
Co-authored-by: Oliver Holworthy <[email protected]>
  • Loading branch information
3 people authored Oct 7, 2022
1 parent b7a2d6e commit 14a18dc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -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 .
Expand Down

0 comments on commit 14a18dc

Please sign in to comment.