scikit-learn>=1.0.0
pandas>=1.0.0

[all]
scikit-learn>=1.0.0
pandas>=1.0.0
sentence-transformers>=2.2.2
sense2vec==2.0.0
timm>=0.6.7

[dev]
scikit-learn>=1.0.0
pandas>=1.0.0
sentence-transformers>=2.2.2
sense2vec==2.0.0
timm>=0.6.7
mkdocs==1.1
mkdocs-material==4.6.3
mkdocstrings==0.8.0
mktestdocs==0.1.2
interrogate>=1.5.0
flake8>=3.6.0
pytest>=4.0.2
black>=19.3b0
pre-commit>=2.2.0
mktestdocs==0.1.2
datasets==2.8.0
matplotlib==3.4.3

[pytorch]
torch>=1.12.0
scikit-learn>=1.0.0
pandas>=1.0.0

[sense2vec]
sense2vec==2.0.0
scikit-learn>=1.0.0
pandas>=1.0.0

[sentence-tfm]
sentence-transformers>=2.2.2
scikit-learn>=1.0.0
pandas>=1.0.0

[text]
sentence-transformers>=2.2.2
sense2vec==2.0.0
scikit-learn>=1.0.0
pandas>=1.0.0

[vision]
timm>=0.6.7
scikit-learn>=1.0.0
pandas>=1.0.0
