adding numpy to the integration tests (#28831)

ivy-llc · Oct 1, 2024 · eb7acb4 · eb7acb4
1 parent 264dbaf
commit eb7acb4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/.github/workflows/integration-tests.yml b/.github/workflows/integration-tests.yml
@@ -16,7 +16,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        target : [ jax, tensorflow ]
+        target : [ jax, numpy, tensorflow ]
     steps:
       - name: Checkout Ivy 🛎
         uses: actions/checkout@v3

diff --git a/ivy_tests/test_integrations/helpers.py b/ivy_tests/test_integrations/helpers.py
@@ -10,7 +10,7 @@
 jax.config.update("jax_enable_x64", True)
 
 jax_kornia = ivy.transpile(kornia, source="torch", target="jax")
-# np_kornia = ivy.transpile(kornia, source="torch", target="numpy")
+np_kornia = ivy.transpile(kornia, source="torch", target="numpy")
 tf_kornia = ivy.transpile(kornia, source="torch", target="tensorflow")
 
 

diff --git a/ivy_tests/test_integrations/test_kornia.py b/ivy_tests/test_integrations/test_kornia.py
@@ -3,7 +3,7 @@
 from helpers import _test_function
 import kornia
 import torch
-
+import pytest 
 
 def test_rgb_to_grayscale(target_framework, backend_compile):
     trace_args = (torch.rand(1, 3, 4, 4),)
@@ -1528,6 +1528,9 @@ def test_determinant_to_polynomial(target_framework, backend_compile):
 
 
 def test_nms3d(target_framework, backend_compile):
+    if target_framework == 'numpy':
+        pytest.skip() # stateful class tests are not supported
+
     trace_args = (
         torch.rand(1, 1, 5, 5, 5),
         (3, 3, 3),