fix: fix dtypes in frontend test for torch.tensor __floordiv__

ivy-llc · Feb 27, 2024 · 70a3bd6 · 70a3bd6
1 parent 5f99f4f
commit 70a3bd6
Showing 1 changed file with 3 additions and 3 deletions.
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
@@ -653,7 +653,7 @@ def test_torch___eq__(
     init_tree="torch.tensor",
     method_name="__floordiv__",
     dtype_and_x=helpers.dtype_and_values(
-        available_dtypes=helpers.get_dtypes("float"),
+        available_dtypes=helpers.get_dtypes("float_and_integer"),
         num_arrays=2,
         large_abs_safety_factor=2.5,
         small_abs_safety_factor=2.5,
@@ -672,12 +672,12 @@ def test_torch___floordiv__(
     input_dtype, x = dtype_and_x
     assume(not np.any(np.isclose(x[1], 0)))
     helpers.test_frontend_method(
-        init_input_dtypes=input_dtype,
+        init_input_dtypes=[input_dtype[0]],
         backend_to_test=backend_fw,
         init_all_as_kwargs_np={
             "data": x[0],
         },
-        method_input_dtypes=input_dtype,
+        method_input_dtypes=[input_dtype[1]],
         method_all_as_kwargs_np={
             "other": x[1],
         },