Unverified Commit e47843c5 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fix reference tests for convert_format_bounding_box (#6860)

* fix reference tests for convert_format_bounding_box

* add check to prevent empty args_kwargs_fns in the future
parent 436ff9a5
...@@ -536,12 +536,12 @@ def sample_inputs_convert_format_bounding_box(): ...@@ -536,12 +536,12 @@ def sample_inputs_convert_format_bounding_box():
def reference_convert_format_bounding_box(bounding_box, old_format, new_format): def reference_convert_format_bounding_box(bounding_box, old_format, new_format):
return torchvision.ops.box_convert( return torchvision.ops.box_convert(
bounding_box, in_fmt=old_format.kernel_name.lower(), out_fmt=new_format.kernel_name.lower() bounding_box, in_fmt=old_format.name.lower(), out_fmt=new_format.name.lower()
) ).to(bounding_box.dtype)
def reference_inputs_convert_format_bounding_box(): def reference_inputs_convert_format_bounding_box():
for args_kwargs in sample_inputs_convert_color_space_image_tensor(): for args_kwargs in sample_inputs_convert_format_bounding_box():
if len(args_kwargs.args[0].shape) == 2: if len(args_kwargs.args[0].shape) == 2:
yield args_kwargs yield args_kwargs
......
...@@ -38,6 +38,10 @@ ignore_jit_warning_no_profile = pytest.mark.filterwarnings( ...@@ -38,6 +38,10 @@ ignore_jit_warning_no_profile = pytest.mark.filterwarnings(
def make_info_args_kwargs_params(info, *, args_kwargs_fn, test_id=None): def make_info_args_kwargs_params(info, *, args_kwargs_fn, test_id=None):
args_kwargs = list(args_kwargs_fn(info)) args_kwargs = list(args_kwargs_fn(info))
if not args_kwargs:
raise pytest.UsageError(
f"Couldn't collect a single `ArgsKwargs` for `{info.id}`{f' in {test_id}' if test_id else ''}"
)
idx_field_len = len(str(len(args_kwargs))) idx_field_len = len(str(len(args_kwargs)))
return [ return [
pytest.param( pytest.param(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment