mirror of
https://github.com/elastic/eland.git
synced 2025-07-11 00:02:14 +08:00
41 lines
611 B
Plaintext
41 lines
611 B
Plaintext
#
|
|
# Basic requirements
|
|
#
|
|
elasticsearch>=8.3,<9
|
|
pandas>=1.5,<2
|
|
matplotlib>=3.6
|
|
numpy>=1.2.0,<2
|
|
tqdm<5
|
|
|
|
#
|
|
# Extras
|
|
#
|
|
scikit-learn>=1.3,<2
|
|
xgboost>=0.90,<2
|
|
lightgbm>=2,<4
|
|
|
|
# PyTorch doesn't support Python 3.11 yet (pytorch/pytorch#86566)
|
|
|
|
# Elasticsearch uses v1.13.1 of PyTorch
|
|
torch>=1.13.1,<2.0; python_version<'3.11'
|
|
# Versions known to be compatible with PyTorch 1.13.1
|
|
sentence-transformers>=2.1.0,<=2.2.2; python_version<'3.11'
|
|
transformers[torch]>=4.31.0,<=4.33.2; python_version<'3.11'
|
|
|
|
#
|
|
# Testing
|
|
#
|
|
pytest>=5.2.1
|
|
pytest-mock
|
|
pytest-cov
|
|
nbval
|
|
shap==0.43.0
|
|
|
|
#
|
|
# Docs
|
|
#
|
|
nox
|
|
numpydoc>=0.9.0
|
|
mypy
|
|
build
|
|
twine |