mirror of
https://github.com/elastic/eland.git
synced 2025-07-11 00:02:14 +08:00
36 lines
491 B
Plaintext
36 lines
491 B
Plaintext
#
|
|
# Basic requirements
|
|
#
|
|
elasticsearch>=8,<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)
|
|
sentence-transformers>=2.1.0,<3; python_version<'3.10'
|
|
torch>=1.9.0,<2; python_version<'3.10'
|
|
transformers[torch]>=4.12.0,<5; python_version<'3.10'
|
|
|
|
#
|
|
# Testing
|
|
#
|
|
pytest>=5.2.1
|
|
pytest-mock
|
|
pytest-cov
|
|
nbval
|
|
|
|
#
|
|
# Docs
|
|
#
|
|
nox
|
|
numpydoc>=0.9.0
|
|
mypy |