From 65114291fbbe33016e9629289cdd696e8df1da4a Mon Sep 17 00:00:00 2001 From: Jackson McClintock Date: Sat, 13 Apr 2024 12:44:19 -0400 Subject: [PATCH] fix: fix available dtypes in frontend tests for torch.tensor --- .../test_ivy/test_frontends/test_torch/test_tensor.py | 8 ++++++-- 1 file changed, 6 insertions(+), 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 index 2da6fccbc59f3..2a535136e56c4 100644 --- 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,