elasticsearch<9,>=8.3
pandas<2,>=1.5
matplotlib>=3.6
numpy<2,>=1.2.0
packaging

[all]
torch==2.1.2
scikit-learn<1.4,>=1.3
transformers[torch]<4.36.0,>=4.31.0
lightgbm<4,>=2
sentence-transformers<=2.3.1,>=2.1.0
xgboost<2,>=0.90

[lightgbm]
lightgbm<4,>=2

[pytorch]
torch==2.1.2
sentence-transformers<=2.3.1,>=2.1.0
transformers[torch]<4.36.0,>=4.31.0

[scikit-learn]
scikit-learn<1.4,>=1.3

[xgboost]
xgboost<2,>=0.90
