nvidia-pyindex
pybind11

-r base.txt
-r dev.txt

tritonclient[all]
torch

pytest>=5
pytest-cov>=2
asvdb@git+https://github.com/rapidsai/asvdb.git
testbook
ipykernel

# needed to make test_s3 work
pytest-xdist
s3fs>=2021.4
aiobotocore>=1.3.3

# required for synthetic data `merlin.datasets` and notebook tests using merlin models
merlin-models[tensorflow]@git+https://github.com/NVIDIA-Merlin/models.git

# needed to run notebook tests
nest-asyncio
scikit-learn>=0.20
graphviz>=0.16


