diff --git a/requirements-dev.txt b/requirements-dev.txt index 4135f25..490c93b 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -4,7 +4,9 @@ elasticsearch>=8.3,<9 pandas>=1.5,<2 matplotlib>=3.6 -numpy<1.24 +# Shap is incompatible with NumPy >= 1.24 (elastic/eland#539) +# Fix NumPy to a known good range of versions +numpy>=1.2.0,<1.24 tqdm<5 # diff --git a/requirements.txt b/requirements.txt index f4fbfda..a62f2ae 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,4 +4,6 @@ elasticsearch>=8.3,<9 pandas>=1.5,<2 matplotlib>=3.6 -numpy<2 +# Shap is incompatible with NumPy >= 1.24 (elastic/eland#539) +# Fix NumPy to a known good range of versions +numpy>=1.2.0,<1.24 diff --git a/setup.py b/setup.py index 7894754..63ab3c7 100644 --- a/setup.py +++ b/setup.py @@ -85,7 +85,7 @@ setup( "elasticsearch>=8.3,<9", "pandas>=1.5,<2", "matplotlib>=3.6", - "numpy<1.24", + "numpy>=1.2.0,<1.24", ], scripts=["bin/eland_import_hub_model"], python_requires=">=3.8",