fixed expm1 method in the PyTorch frontend (#21745)

Co-authored-by: NripeshN <[email protected]>
ivy-llc · Dec 23, 2023 · b2389e0 · b2389e0
1 parent 5f58f81
commit b2389e0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ivy/functional/frontends/torch/tensor.py b/ivy/functional/frontends/torch/tensor.py
@@ -1386,8 +1386,8 @@ def cov(self, /, *, correction=1, fweights=None, aweights=None):
     def exp(self):
         return torch_frontend.exp(self)
 
-    @with_unsupported_dtypes(
-        {"2.1.2 and below": ("bfloat16", "float16", "complex")}, "torch"
+    @with_supported_dtypes(
+        {"2.1.2 and below": ("bfloat16", "float32", "float64")}, "torch"
     )
     def expm1(self):
         return torch_frontend.expm1(self)

diff --git a/ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py b/ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py
@@ -6533,7 +6533,7 @@ def test_torch_expand_as(
     init_tree="torch.tensor",
     method_name="expm1",
     dtype_and_x=helpers.dtype_and_values(
-        available_dtypes=helpers.get_dtypes("numeric"),
+        available_dtypes=helpers.get_dtypes("valid"),
     ),
 )
 def test_torch_expm1(