From d26c0d0caec9b94ead11b55f7170e42050c25bf2 Mon Sep 17 00:00:00 2001 From: Yusha Arif Date: Tue, 1 Oct 2024 07:24:49 +0000 Subject: [PATCH 1/3] adding numpy to the integration tests --- .github/workflows/integration-tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/integration-tests.yml b/.github/workflows/integration-tests.yml index fa61e2f40e60e..32d63efeccb69 100644 --- 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 From eebe31603815e03735c6c580c0a567627d6d26f3 Mon Sep 17 00:00:00 2001 From: Yusha Arif Date: Tue, 1 Oct 2024 10:30:35 +0000 Subject: [PATCH 2/3] fix: adding lazy transpilation for `np_kornia` --- ivy_tests/test_integrations/helpers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ivy_tests/test_integrations/helpers.py b/ivy_tests/test_integrations/helpers.py index 9d9c7c699a78e..408aeeb229040 100644 --- 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") From e00180e250ec18bf7ae03d3a63b4b9c44ee9685c Mon Sep 17 00:00:00 2001 From: Yusha Arif Date: Tue, 1 Oct 2024 11:10:18 +0000 Subject: [PATCH 3/3] skipping `test_nms3d` test for `numpy` --- ivy_tests/test_integrations/test_kornia.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ivy_tests/test_integrations/test_kornia.py b/ivy_tests/test_integrations/test_kornia.py index 387189aeff75f..88817fa2bf6a5 100644 --- 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),