Skip to content

Commit 2199b55

Browse files
committed
fix: test_torch_baddbmm_
1 parent caba23d commit 2199b55

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4034,7 +4034,7 @@ def test_torch_baddbmm(
40344034
helpers.test_frontend_method(
40354035
init_input_dtypes=input_dtype,
40364036
backend_to_test=backend_fw,
4037-
init_all_as_kwargs_np={"data": x[0]},
4037+
init_all_as_kwargs_np={"data": x},
40384038
method_input_dtypes=input_dtype,
40394039
method_all_as_kwargs_np={
40404040
"batch1": batch1,
@@ -4069,7 +4069,6 @@ def test_torch_baddbmm(
40694069
allow_subnormal=False,
40704070
allow_infinity=False,
40714071
),
4072-
test_inplace=st.just(True),
40734072
)
40744073
def test_torch_baddbmm_(
40754074
dtype_and_matrices,
@@ -4086,7 +4085,7 @@ def test_torch_baddbmm_(
40864085
helpers.test_frontend_method(
40874086
init_input_dtypes=input_dtype,
40884087
backend_to_test=backend_fw,
4089-
init_all_as_kwargs_np={"data": x[0]},
4088+
init_all_as_kwargs_np={"data": x},
40904089
method_input_dtypes=input_dtype,
40914090
method_all_as_kwargs_np={
40924091
"batch1": batch1,

0 commit comments

Comments
 (0)