test: Add NumPy < 2 constraint

astefanutti · astefanutti · commit 3c13debb84d2 · 2024-06-28T16:06:10.000+02:00
diff --git a/test/e2e/mnist_pip_requirements.txt b/test/e2e/mnist_pip_requirements.txt
@@ -1,3 +1,4 @@
 pytorch_lightning==1.9.5
 torchmetrics==0.9.1
 torchvision==0.12.0
+numpy<=2.0.0
diff --git a/test/e2e/mnist_rayjob_raycluster_test.go b/test/e2e/mnist_rayjob_raycluster_test.go
@@ -316,6 +316,7 @@ func constructRayJob(_ Test, namespace *corev1.Namespace, rayCluster *rayv1.RayC
     - pytorch_lightning==1.9.5
     - torchmetrics==0.9.1
     - torchvision==0.12.0
+    - numpy<=2.0.0
   env_vars:
     MNIST_DATASET_URL: "` + GetMnistDatasetURL() + `"
     PIP_INDEX_URL: "` + GetPipIndexURL() + `"
