Merge pull request #733 from RasmusOrsoe/fix_tests

Fix unit tests
graphnet-team · Jul 20, 2024 · 652f194 · 652f194
2 parents 72c38d5 + b0d24b8
commit 652f194
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -109,6 +109,7 @@ jobs:
           pip show torch-cluster
           pip show torch-sparse
           pip show torch-scatter
+          pip show numpy
       - name: Run unit tests and generate coverage report
         run: |
           set -o pipefail  # To propagate exit code from pytest

diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
     "configupdater",
     "dill>=0.3",
     "matplotlib>=3.5",
-    "numpy>=1.21",
+    "numpy>=1.22,<2.0",
     "pandas>=1.3",
     "pyarrow",
     "pydantic",