fix(frontend): add @to_ivy_arrays_and_back to torch.polar

dettmerramon · Sep 15, 2023 · bbacc91 · bbacc91
1 parent 13113f2
commit bbacc91
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/ivy/functional/frontends/torch/creation_ops.py b/ivy/functional/frontends/torch/creation_ops.py
@@ -256,6 +256,7 @@ def ones_like_v_0p4p0_and_above(
 
 
 @with_supported_dtypes({"2.0.1 and below": ("float32", "float64")}, "torch")
+@to_ivy_arrays_and_back
 def polar(
     abs,
     angle,

diff --git a/ivy_tests/test_ivy/test_frontends/test_torch/test_creation_ops.py b/ivy_tests/test_ivy/test_frontends/test_torch/test_creation_ops.py
@@ -181,6 +181,7 @@ def test_complex(
 @handle_frontend_test(
     fn_tree="torch.polar",
     dtype_and_x=helpers.dtype_and_values(available_dtypes=helpers.get_dtypes("float")),
+    test_with_out=st.just(False),
 )
 def test_polar(
     *,
-Original file line number
+Diff line change
@@ Expand Up / @@ -256,6 +256,7 @@ def ones_like_v_0p4p0_and_above( @@
     @with_supported_dtypes({"2.0.1 and below": ("float32", "float64")}, "torch")
+    @to_ivy_arrays_and_back
     def polar(
         abs,
         angle,
@@ Expand Down @@