From 8ba04b75332b6949d283bafbf7203711d385b154 Mon Sep 17 00:00:00 2001
From: Bernie Beckerman <bbeckerman@meta.com>
Date: Wed, 7 Aug 2024 13:01:44 -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

fbshipit-source-id: 8fda1ee98bde1000b970d45b75c75a735e205a08
---
 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"]