From 4627272585cf7881771a45f1ed803523d336396e Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@gmail.com>
Date: Tue, 3 Sep 2024 12:09:08 +0200
Subject: [PATCH] [ci] pin numpy to < 2 on win

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 0e4a80ae7e..6f4d5e1e2f 100644
--- a/setup.py
+++ b/setup.py
@@ -64,7 +64,8 @@
 REQUIRED_PKGS = [
     "torch>=1.4.0",
     "transformers>=4.31.0",
-    "numpy>=1.18.2",
+    "numpy>=1.18.2;platform_system!='Windows'",
+    "numpy<2;platform_system=='Windows'",
     "accelerate",
     "datasets",
     "tyro>=0.5.11",