pandas<3.0.0,>=2.0.0
pyarrow<15.0.0,>=11.0.0
scikit-learn<2.0.0,>=1.2.0
pydantic<3.0.0,>=2.0.0
pyyaml<7.0,>=6.0
jinja2<4.0.0,>=3.1.2
thefuzz<0.21.0,>=0.20.0

[active]
cleanlab<3.0.0,>=2.3.0

[additional_models]
bertopic<0.16.0,>=0.15.0
top2vec<2.0.0,>=1.0.27
nmf-topic-modeling<0.3.0,>=0.2.0
contextualized-topic-models<3.0.0,>=2.5.0

[clustering]
hdbscan<0.9.0,>=0.8.29

[dev]
black<24.0.0,>=23.3.0
ruff<0.1.0,>=0.0.265
mypy<2.0.0,>=1.3.0

[docs]
sphinx<8.0.0,>=7.0.0
sphinx-rtd-theme<2.0.0,>=1.2.0
jupyter<2.0.0,>=1.0.0

[embeddings]
sentence-transformers<3.0.0,>=2.2.2
transformers<5.0.0,>=4.28.0
torch<3.0.0,>=2.0.0

[embeddings-gpu]
sentence-transformers<3.0.0,>=2.2.2
transformers<5.0.0,>=4.28.0
torch<3.0.0,>=2.0.0
accelerate<1.0.0,>=0.20.0
bitsandbytes<1.0.0,>=0.41.0

[full]
sentence-transformers<3.0.0,>=2.2.2
transformers<5.0.0,>=4.28.0
torch<3.0.0,>=2.0.0
gensim<5.0.0,>=4.3.0
plotly<6.0.0,>=5.14.0
umap-learn<0.6.0,>=0.5.3
hdbscan<0.9.0,>=0.8.29
spacy<4.0.0,>=3.5.0
cleanlab<3.0.0,>=2.3.0
polars<1.15.0,>=1.11.0
bertopic<0.16.0,>=0.15.0
top2vec<2.0.0,>=1.0.27

[full-gpu]
sentence-transformers<3.0.0,>=2.2.2
transformers<5.0.0,>=4.28.0
torch<3.0.0,>=2.0.0
accelerate<1.0.0,>=0.20.0
bitsandbytes<1.0.0,>=0.41.0
gensim<5.0.0,>=4.3.0
plotly<6.0.0,>=5.14.0
umap-learn<0.6.0,>=0.5.3
hdbscan<0.9.0,>=0.8.29
spacy<4.0.0,>=3.5.0
cleanlab<3.0.0,>=2.3.0
polars<1.15.0,>=1.11.0
bertopic<0.16.0,>=0.15.0
top2vec<2.0.0,>=1.0.27

[lda]
gensim<5.0.0,>=4.3.0

[nlp]
spacy<4.0.0,>=3.5.0

[optimization]
polars<1.15.0,>=1.11.0

[test]
pytest<8.0.0,>=7.3.1
pytest-cov<5.0.0,>=4.1.0
hypothesis<7.0.0,>=6.75.0

[viz]
plotly<6.0.0,>=5.14.0
umap-learn<0.6.0,>=0.5.3
