mirror of
https://github.com/elastic/eland.git
synced 2025-07-11 00:02:14 +08:00
0576114a1d
Closes #503 Note: I also had to fix the Sphinx version to 5.3.0 since, starting from 6.0, Sphinx suffers from a TypeError bug, which causes a CI failure.
39 lines
590 B
Plaintext
39 lines
590 B
Plaintext
#
|
|
# Basic requirements
|
|
#
|
|
elasticsearch>=8.3,<9
|
|
pandas>=1.5
|
|
matplotlib>=3.6
|
|
numpy<2
|
|
tqdm<5
|
|
|
|
#
|
|
# Extras
|
|
#
|
|
scikit-learn>=0.22.1,<2
|
|
xgboost>=0.90,<2
|
|
lightgbm>=2,<4
|
|
|
|
# PyTorch doesn't support Python 3.10 yet (pytorch/pytorch#66424)
|
|
|
|
# Elasticsearch uses v1.11.0 of PyTorch
|
|
torch>=1.11.0,<1.12.0; python_version<'3.10'
|
|
# Versions known to be compatible with torch 1.11
|
|
sentence-transformers>=2.1.0,<=2.2.2; python_version<'3.10'
|
|
transformers[torch]>=4.12.0,<=4.20.1; python_version<'3.10'
|
|
|
|
#
|
|
# Testing
|
|
#
|
|
pytest>=5.2.1
|
|
pytest-mock
|
|
pytest-cov
|
|
nbval
|
|
shap==0.41.0
|
|
|
|
#
|
|
# Docs
|
|
#
|
|
nox
|
|
numpydoc>=0.9.0
|
|
mypy |