apipkg
colorlog
huggingface-hub
joblib>=0.14.1
numpy<=1.26.4
pydantic>=2.0
pyod>=1.0.0
pythresh>=0.3.0
scikit-learn>=0.20.0
scipy>=1.3.1
torch
transformers
tokenizers
umap-learn

[dev]
pytest
pytest-cov
flake8
mypy
typed-ast
pydocstringformatter
ruff
autopep8
isort
openai<2.0.0,>=1.10.0
cohere<6.00,>=5.00
boto3
botocore
fastembed
google-cloud-aiplatform
mlflow
llama-index-embeddings-huggingface
llama-index
faiss-cpu
langchain-community
sentence-transformers
networkx
pygod
torch_geometric
torch_scatter
torch_sparse
torch_cluster
torch_spline_conv

[graph]
networkx
pygod
torch_geometric
torch_scatter
torch_sparse
torch_cluster
torch_spline_conv

[notebook]
mlflow
pandas
faiss-cpu
langchain-community
sentence-transformers
matplotlib
plotly
datasets
pygod
torch_geometric
