From 1e2e4bcb4761ba6107fb565982f6cc2b951cbeb5 Mon Sep 17 00:00:00 2001
From: Kate Baumli <baumli@google.com>
Date: Tue, 28 Jun 2022 04:33:15 -0700
Subject: [PATCH] Pin numpy version <1.23 until new jax version is released,
 fixing bug that makes mpo_ops_test fail.

PiperOrigin-RevId: 457697892
---
 requirements/requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index 418f715..af44379 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -4,4 +4,4 @@ distrax>=0.0.2
 dm_env
 jax>=0.1.55
 jaxlib>=0.1.37
-numpy>=1.18.0
+numpy>=1.18.0,<1.23