mirror of
https://github.com/elastic/eland.git
synced 2025-07-11 00:02:14 +08:00
36bbbe0bdb
PyTorch models traced in version 1.13 of PyTorch cannot be evaluated in version 1.9 or earlier. With this upgrade Eland becomes incompatible with pre 8.7 Elasticsearch and will refuse to upload a model to the cluster. In this scenario either upgrade Elasticsearch or use an earlier version of Eland.
39 lines
597 B
Plaintext
39 lines
597 B
Plaintext
#
|
|
# Basic requirements
|
|
#
|
|
elasticsearch>=8.3,<9
|
|
pandas>=1.5,<2
|
|
matplotlib>=3.6
|
|
numpy<1.24
|
|
tqdm<5
|
|
|
|
#
|
|
# Extras
|
|
#
|
|
scikit-learn>=0.22.1,<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.12.0,<=4.27.4; python_version<'3.11'
|
|
|
|
#
|
|
# Testing
|
|
#
|
|
pytest>=5.2.1
|
|
pytest-mock
|
|
pytest-cov
|
|
nbval
|
|
shap==0.41.0
|
|
|
|
#
|
|
# Docs
|
|
#
|
|
nox
|
|
numpydoc>=0.9.0
|
|
mypy |