From b7bddf03fdc29a1223b00f7f23ef00430adb3b40 Mon Sep 17 00:00:00 2001 From: Bernie Beckerman Date: Wed, 7 Aug 2024 12:08:58 -0700 Subject: [PATCH] pin numpy to < 2.0 until tensorboard cuts a release (#2641) Summary: Pull Request resolved: https://github.com/facebook/Ax/pull/2641 The latest TB version hasn't incorporated numpy2.0-compatible changes (https://github.com/tensorflow/tensorboard/issues/6869#issuecomment-2273718763), and recently unpinned botorch (D60735276) means Ax CI is now failing. This pins numpy to < 2.0 to prevent issues in CI. Differential Revision: D60917879 --- setup.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/setup.py b/setup.py index 0b236fe2299..39e99f7dc63 100644 --- a/setup.py +++ b/setup.py @@ -46,6 +46,9 @@ "torchvision>=0.5.0", "nbconvert", "jupyter-client==6.1.12", + # Replace with `tensorboard >= x.x` once tb cuts a release. + # https://github.com/tensorflow/tensorboard/issues/6869#issuecomment-2273718763 + "numpy<2.0", ] MYSQL_REQUIRES = ["SQLAlchemy==1.4.17"]