numpy < 2 for onnx

pytorch · Aug 10, 2024 · d1e6abf · d1e6abf
1 parent e0b11b2
commit d1e6abf
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -127,7 +127,7 @@ jobs:
         echo '::endgroup::'
 
         echo '::group::Install testing utilities'
-        pip install --progress-bar=off pytest
+        pip install --progress-bar=off pytest "numpy<2"
         echo '::endgroup::'
 
         echo '::group::Run ONNX tests'

diff --git a/test/test_onnx.py b/test/test_onnx.py
@@ -422,7 +422,6 @@ def get_test_images(self) -> Tuple[List[torch.Tensor], List[torch.Tensor]]:
             [self.get_image("fakedata/logos/rgb_pytorch.png", (250, 380))],
         )
 
-    @pytest.mark.skip("Too long on CPU")
     def test_faster_rcnn(self):
         images, test_images = self.get_test_images()
         dummy_image = [torch.ones(3, 100, 100) * 0.3]