From 7820a31256ab89c19af354b6343f4134567b63ae Mon Sep 17 00:00:00 2001 From: David Kyle Date: Mon, 22 May 2023 10:47:06 +0100 Subject: [PATCH] Limit NumPy to a range of versions and note why (#540) --- requirements-dev.txt | 4 +++- requirements.txt | 4 +++- setup.py | 2 +- 3 files changed, 7 insertions(+), 3 deletions(-) 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",