mirror of
https://github.com/elastic/eland.git
synced 2025-07-11 00:02:14 +08:00
0eb36faa5b
Elasticsearch uses v1.11 of PyTorch. Models created with the latest PyTorch release (v1.12) are not compatible with v1.11. This pins the PyTorch version to 1.11 to prevent the incompatibility. The version of the Elasticsearch Python client is now required to be >= Eland. All users of Eland for importing NLP models should upgrade.
39 lines
601 B
Plaintext
39 lines
601 B
Plaintext
#
|
|
# Basic requirements
|
|
#
|
|
elasticsearch>=8.3,<9
|
|
pandas>=1.2,<2
|
|
matplotlib<4
|
|
numpy<2
|
|
tqdm<5
|
|
|
|
#
|
|
# Extras
|
|
#
|
|
scikit-learn>=0.22.1,<2
|
|
xgboost>=0.90,<2
|
|
scikit-learn>=0.22.1,<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
|
|
|
|
#
|
|
# Docs
|
|
#
|
|
nox
|
|
numpydoc>=0.9.0
|
|
mypy |