fix: Fixed tf.math.argmin for torch, jax, and paddle backends (#28342)

ivy-llc · Feb 20, 2024 · 503c29c · 503c29c
1 parent 704c571
commit 503c29c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/ivy/functional/backends/torch/searching.py b/ivy/functional/backends/torch/searching.py
@@ -41,7 +41,15 @@ def argmax(
     return ret
 
 
-@with_unsupported_dtypes({"2.2 and below": ("complex",)}, backend_version)
+@with_unsupported_dtypes(
+    {
+        "2.2 and below": (
+            "complex",
+            "bool",
+        )
+    },
+    backend_version,
+)
 def argmin(
     x: torch.Tensor,
     /,

diff --git a/ivy_tests/test_ivy/test_frontends/test_tensorflow/test_math.py b/ivy_tests/test_ivy/test_frontends/test_tensorflow/test_math.py
@@ -286,7 +286,7 @@ def test_tensorflow_argmin(
     on_device,
     output_type,
 ):
-    input_dtype, x, axis = dtype_and_x
+    input_dtype, x, axis, *_ = dtype_and_x
     if isinstance(axis, tuple):
         axis = axis[0]
     helpers.test_frontend_function(