Skip to content

Commit 8656823

Browse files
committed
fix: test_torch_renorm
1 parent 034bd15 commit 8656823

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

ivy/functional/frontends/torch/miscellaneous_ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -438,7 +438,7 @@ def ravel(input):
438438
return ivy.reshape(input, (-1,))
439439

440440

441-
@with_unsupported_dtypes({"2.2 and below": ("float16",)}, "torch")
441+
@with_unsupported_dtypes({"2.2 and below": ("bfloat16", "float16")}, "torch")
442442
@to_ivy_arrays_and_back
443443
def renorm(input, p, dim, maxnorm, *, out=None):
444444
# Torch hardcodes this magic number

ivy_tests/test_ivy/test_frontends/test_torch/test_miscellaneous_ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1548,7 +1548,7 @@ def test_torch_ravel(
15481548
fn_tree="torch.renorm",
15491549
dtype_and_values=helpers.dtype_and_values(
15501550
shape=st.shared(helpers.get_shape(min_num_dims=2), key="shape"),
1551-
available_dtypes=helpers.get_dtypes("numeric"),
1551+
available_dtypes=helpers.get_dtypes("float_and_complex"),
15521552
max_value=1e4,
15531553
min_value=-1e4,
15541554
),

0 commit comments

Comments
 (0)