sentence-transformers<5.0.0,>4.0.0
transformers[torch]>=4.34.0
optuna>4.2.0
wandb>0.19.4
torch>=2.0.0
scikit-learn==1.6.1
pandas>=2.0.0
pandarallel==1.6.5
numpy>=1.26.4
hydra-core>=1.3.2
einops==0.8.0

[benchmark]
sqlalchemy==2.0.37
psycopg2-binary==2.9.9
pgvector==0.3.6

[data]
pyarrow==19.0.0
safe-mol==0.1.13
rdkit==2024.9.4
opencv-python==4.11.0.86
beautifulsoup4==4.12.3
fickling==0.1.3

[dev]
pytest
pytest-cov
pytest-xdist
ipykernel
ipywidgets
build
ruff
ninja==1.11.1.3
