-r base.txt
-r dev.txt

# This contains common libraries for testing.

# NOTE: You should pip install requirements/test-[cpu|gpu].txt for device-specific test
# requirements, which will include the dependencies defined in this file.

pytest>=5
pytest-cov>=2
scikit-learn<1.2
asvdb@git+https://github.com/rapidsai/asvdb.git
testbook==0.4.2

# packages necessary to run tests and push PRs
tritonclient
feast==0.19.4
xgboost==1.6.2
implicit==0.6.0

# TODO: do we need more of these?
# https://github.com/NVIDIA-Merlin/Merlin/blob/a1cc48fe23c4dfc627423168436f26ef7e028204/ci/dockerfile.ci#L13-L18
