numpy>=1.17.0
transformers<4.19
tqdm>=4.27
betterproto<2.0.0
pyarrow>=1.0
tensorflow-metadata

[all]
numpy>=1.17.0
transformers<4.19
tqdm>=4.27
betterproto<2.0.0
pyarrow>=1.0
tensorflow-metadata
torch>=1.0
torchmetrics>=0.10.0
nvtabular
merlin-core>=0.2.0
merlin-dataloader>=0.0.2
Sphinx<3.6
sphinx_rtd_theme==1.0.0
sphinx-multiversion@ git+https://github.com/mikemckiernan/sphinx-multiversion.git
sphinxcontrib-copydirs@ git+https://github.com/mikemckiernan/sphinxcontrib-copydirs.git
sphinx-external-toc<0.4
ipython==8.1.1
jinja2<3.1
markupsafe==2.0.1
natsort==8.0.1
myst-nb<0.14
linkify-it-py<1.1
check-manifest
bandit

[base]
numpy>=1.17.0
transformers<4.19
tqdm>=4.27
betterproto<2.0.0
pyarrow>=1.0
tensorflow-metadata

[dataloader]
merlin-core>=0.2.0
merlin-dataloader>=0.0.2

[dev]
check-manifest
bandit

[docs]
Sphinx<3.6
sphinx_rtd_theme==1.0.0
sphinx-multiversion@ git+https://github.com/mikemckiernan/sphinx-multiversion.git
sphinxcontrib-copydirs@ git+https://github.com/mikemckiernan/sphinxcontrib-copydirs.git
sphinx-external-toc<0.4
ipython==8.1.1
jinja2<3.1
markupsafe==2.0.1
natsort==8.0.1
myst-nb<0.14
linkify-it-py<1.1

[nvtabular]
nvtabular

[pytorch]
torch>=1.0
torchmetrics>=0.10.0
