From b16c72ae3906b801be6726b4f1737a2584ec3309 Mon Sep 17 00:00:00 2001
From: Charlie Blake <charlieb@graphcore.ai>
Date: Wed, 6 Nov 2024 16:13:09 +0000
Subject: [PATCH] fix numpy at <2.0.0

---
 pyproject.toml       | 2 +-
 requirements-dev.txt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 86af2d9..5997432 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ dependencies = [
   "datasets",
   "docstring-parser",
   "einops",
-  "numpy",
+  "numpy<2.0.0",
   "seaborn",
   "tabulate",
   "torch>=2.2",
diff --git a/requirements-dev.txt b/requirements-dev.txt
index 50df8b8..f893d36 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -6,7 +6,7 @@
 datasets==3.1.0
 docstring-parser==0.16
 einops==0.8.0
-numpy==2.1.3
+numpy==1.26.4
 seaborn==0.13.2
 tabulate==0.9.0
 torch==2.5.1+cpu