From a59eb2a3cdf4e298e5c605fe14052e67331cb7e0 Mon Sep 17 00:00:00 2001
From: Imri Paran <imri.paran@gmail.com>
Date: Tue, 20 Aug 2024 12:19:05 +0200
Subject: [PATCH] fix: pin numpy version (#17487)

---
 ingestion/setup.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/ingestion/setup.py b/ingestion/setup.py
index e4d76b6ddac..9774dac0d10 100644
--- a/ingestion/setup.py
+++ b/ingestion/setup.py
@@ -65,6 +65,7 @@ COMMONS = {
         VERSIONS["boto3"],
         VERSIONS["pandas"],
         VERSIONS["pyarrow"],
+        VERSIONS["numpy"],
         # python-snappy does not work well on 3.11 https://github.com/aio-libs/aiokafka/discussions/931
         # Using this as an alternative
         "cramjam~=2.7",
@@ -161,7 +162,7 @@ plugins: Dict[str, Set[str]] = {
         VERSIONS["numpy"],
         "sqlalchemy-bigquery>=1.2.2",
     },
-    "bigtable": {"google-cloud-bigtable>=2.0.0", VERSIONS["pandas"]},
+    "bigtable": {"google-cloud-bigtable>=2.0.0", VERSIONS["pandas"], VERSIONS["numpy"]},
     "clickhouse": {
         "clickhouse-driver~=0.2",
         "clickhouse-sqlalchemy~=0.2",
@@ -259,7 +260,7 @@ plugins: Dict[str, Set[str]] = {
         VERSIONS["giturlparse"],
     },
     "mlflow": {"mlflow-skinny>=2.3.0"},
-    "mongo": {VERSIONS["mongo"], VERSIONS["pandas"]},
+    "mongo": {VERSIONS["mongo"], VERSIONS["pandas"], VERSIONS["numpy"]},
     "couchbase": {"couchbase~=4.1"},
     "mssql": {
         "sqlalchemy-pytds~=0.3",
@@ -314,6 +315,7 @@ plugins: Dict[str, Set[str]] = {
     "pii-processor": {
         VERSIONS["spacy"],
         VERSIONS["pandas"],
+        VERSIONS["numpy"],
         "presidio-analyzer==2.2.32",
     },
 }