fix: fix available dtypes in frontend tests for torch.tensor

ivy-llc · Apr 13, 2024 · 6511429 · 6511429
1 parent b600b09
commit 6511429
Showing 1 changed file with 6 additions and 2 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
@@ -476,7 +476,9 @@ def test_torch___add__(
     init_tree="torch.tensor",
     method_name="__and__",
     dtype_and_x=helpers.dtype_and_values(
-        available_dtypes=st.one_of(st.just(("bool",)), helpers.get_dtypes("integer")),
+        available_dtypes=st.one_of(
+            helpers.get_dtypes("bool"), helpers.get_dtypes("integer")
+        ),
         num_arrays=2,
         min_value=-1e04,
         max_value=1e04,
@@ -773,7 +775,9 @@ def test_torch___gt__(
     init_tree="torch.tensor",
     method_name="__iand__",
     dtype_and_x=helpers.dtype_and_values(
-        available_dtypes=st.one_of(st.just(("bool",)), helpers.get_dtypes("integer")),
+        available_dtypes=st.one_of(
+            helpers.get_dtypes("bool"), helpers.get_dtypes("integer")
+        ),
         num_arrays=2,
         min_value=-1e04,
         max_value=1e04,