fix: CI integration tests (#28802)

ivy-llc · Aug 10, 2024 · 78b4ce7 · 78b4ce7
1 parent fc6fb96
commit 78b4ce7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ivy_tests/test_integrations/helpers.py b/ivy_tests/test_integrations/helpers.py
@@ -9,9 +9,9 @@
 
 jax.config.update("jax_enable_x64", True)
 
-jax_kornia = ivy.transpile(kornia, source="torch", to="jax")
-np_kornia = ivy.transpile(kornia, source="torch", to="numpy")
-tf_kornia = ivy.transpile(kornia, source="torch", to="tensorflow")
+jax_kornia = ivy.graph_transpile(kornia, source="torch", to="jax")
+np_kornia = ivy.graph_transpile(kornia, source="torch", to="numpy")
+tf_kornia = ivy.graph_transpile(kornia, source="torch", to="tensorflow")
 
 
 # Helpers #