From 32e2c67fe3b45ec14a521fd4492674f24f9e758c Mon Sep 17 00:00:00 2001 From: Albert Villanova del Moral <8515462+albertvillanova@users.noreply.github.com> Date: Thu, 12 Sep 2024 10:28:13 +0200 Subject: [PATCH] Pin numpy<2 as required by tensorflow --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index 5a97a847..8a2379ed 100644 --- a/setup.py +++ b/setup.py @@ -79,6 +79,7 @@ "pytest-datadir", "pytest-xdist", # optional dependencies + "numpy<2.0.0", # tensorflow requires numpy < 2 "tensorflow>=2.3,!=2.6.0,!=2.6.1, <=2.10", "torch", # metrics dependencies