diff --git a/src/transformers/models/conditional_detr/image_processing_conditional_detr.py b/src/transformers/models/conditional_detr/image_processing_conditional_detr.py index 1cd2b1eed112..3d12410af744 100644 --- a/src/transformers/models/conditional_detr/image_processing_conditional_detr.py +++ b/src/transformers/models/conditional_detr/image_processing_conditional_detr.py @@ -877,7 +877,9 @@ def __init__( do_pad = kwargs.pop("pad_and_return_pixel_mask") size = size if size is not None else {"shortest_edge": 800, "longest_edge": 1333} - size = {'shortest_edge': size, 'longest_edge': 1333} if isinstance(size, int) else size # Backwards compatibility + size = ( + {"shortest_edge": size, "longest_edge": 1333} if isinstance(size, int) else size + ) # Backwards compatibility size = get_size_dict(size, default_to_square=False) # Backwards compatibility diff --git a/src/transformers/models/deformable_detr/image_processing_deformable_detr.py b/src/transformers/models/deformable_detr/image_processing_deformable_detr.py index b973848015fd..f620fa76fff7 100644 --- a/src/transformers/models/deformable_detr/image_processing_deformable_detr.py +++ b/src/transformers/models/deformable_detr/image_processing_deformable_detr.py @@ -875,7 +875,9 @@ def __init__( do_pad = kwargs.pop("pad_and_return_pixel_mask") size = size if size is not None else {"shortest_edge": 800, "longest_edge": 1333} - size = {'shortest_edge': size, 'longest_edge': 1333} if isinstance(size, int) else size # Backwards compatibility + size = ( + {"shortest_edge": size, "longest_edge": 1333} if isinstance(size, int) else size + ) # Backwards compatibility size = get_size_dict(size, default_to_square=False) # Backwards compatibility diff --git a/src/transformers/models/deformable_detr/image_processing_deformable_detr_fast.py b/src/transformers/models/deformable_detr/image_processing_deformable_detr_fast.py index a829444b2543..8f0380be9d94 100644 --- a/src/transformers/models/deformable_detr/image_processing_deformable_detr_fast.py +++ b/src/transformers/models/deformable_detr/image_processing_deformable_detr_fast.py @@ -353,7 +353,9 @@ def __init__( do_pad = kwargs.pop("pad_and_return_pixel_mask") size = size if size is not None else {"shortest_edge": 800, "longest_edge": 1333} - size = {'shortest_edge': size, 'longest_edge': 1333} if isinstance(size, int) else size # Backwards compatibility + size = ( + {"shortest_edge": size, "longest_edge": 1333} if isinstance(size, int) else size + ) # Backwards compatibility size = get_size_dict(size, default_to_square=False) # Backwards compatibility diff --git a/src/transformers/models/detr/image_processing_detr.py b/src/transformers/models/detr/image_processing_detr.py index 8b1e60315a2a..80ee10970f5d 100644 --- a/src/transformers/models/detr/image_processing_detr.py +++ b/src/transformers/models/detr/image_processing_detr.py @@ -859,7 +859,9 @@ def __init__( do_pad = kwargs.pop("pad_and_return_pixel_mask") size = size if size is not None else {"shortest_edge": 800, "longest_edge": 1333} - size = {'shortest_edge': size, 'longest_edge': 1333} if isinstance(size, int) else size # Backwards compatibility + size = ( + {"shortest_edge": size, "longest_edge": 1333} if isinstance(size, int) else size + ) # Backwards compatibility size = get_size_dict(size, default_to_square=False) # Backwards compatibility diff --git a/src/transformers/models/detr/image_processing_detr_fast.py b/src/transformers/models/detr/image_processing_detr_fast.py index ad6b97d2e4b3..d837cca97b1d 100644 --- a/src/transformers/models/detr/image_processing_detr_fast.py +++ b/src/transformers/models/detr/image_processing_detr_fast.py @@ -362,7 +362,9 @@ def __init__( do_pad = kwargs.pop("pad_and_return_pixel_mask") size = size if size is not None else {"shortest_edge": 800, "longest_edge": 1333} - size = {'shortest_edge': size, 'longest_edge': 1333} if isinstance(size, int) else size # Backwards compatibility + size = ( + {"shortest_edge": size, "longest_edge": 1333} if isinstance(size, int) else size + ) # Backwards compatibility size = get_size_dict(size, default_to_square=False) # Backwards compatibility diff --git a/src/transformers/models/grounding_dino/image_processing_grounding_dino.py b/src/transformers/models/grounding_dino/image_processing_grounding_dino.py index 5c6261e91e15..58b22c8fd08a 100644 --- a/src/transformers/models/grounding_dino/image_processing_grounding_dino.py +++ b/src/transformers/models/grounding_dino/image_processing_grounding_dino.py @@ -883,7 +883,9 @@ def __init__( do_pad = kwargs.pop("pad_and_return_pixel_mask") size = size if size is not None else {"shortest_edge": 800, "longest_edge": 1333} - size = {'shortest_edge': size, 'longest_edge': 1333} if isinstance(size, int) else size # Backwards compatibility + size = ( + {"shortest_edge": size, "longest_edge": 1333} if isinstance(size, int) else size + ) # Backwards compatibility size = get_size_dict(size, default_to_square=False) # Backwards compatibility diff --git a/src/transformers/models/rt_detr/image_processing_rt_detr.py b/src/transformers/models/rt_detr/image_processing_rt_detr.py index e9904f5ccd5d..5b1d1e7128c1 100644 --- a/src/transformers/models/rt_detr/image_processing_rt_detr.py +++ b/src/transformers/models/rt_detr/image_processing_rt_detr.py @@ -453,7 +453,9 @@ def __init__( **kwargs, ) -> None: size = size if size is not None else {"height": 640, "width": 640} - size = {'shortest_edge': size, 'longest_edge': 1333} if isinstance(size, int) else size # Backwards compatibility + size = ( + {"shortest_edge": size, "longest_edge": 1333} if isinstance(size, int) else size + ) # Backwards compatibility size = get_size_dict(size, default_to_square=False) if do_convert_annotations is None: diff --git a/src/transformers/models/yolos/image_processing_yolos.py b/src/transformers/models/yolos/image_processing_yolos.py index 74a9f52e14af..81e760ab261a 100644 --- a/src/transformers/models/yolos/image_processing_yolos.py +++ b/src/transformers/models/yolos/image_processing_yolos.py @@ -791,7 +791,9 @@ def __init__( do_pad = kwargs.pop("pad_and_return_pixel_mask") size = size if size is not None else {"shortest_edge": 800, "longest_edge": 1333} - size = {'shortest_edge': size, 'longest_edge': 1333} if isinstance(size, int) else size # Backwards compatibility + size = ( + {"shortest_edge": size, "longest_edge": 1333} if isinstance(size, int) else size + ) # Backwards compatibility size = get_size_dict(size, default_to_square=False) # Backwards compatibility diff --git a/tests/models/deformable_detr/test_image_processing_deformable_detr.py b/tests/models/deformable_detr/test_image_processing_deformable_detr.py index c9dd88797d52..93882ebef5c6 100644 --- a/tests/models/deformable_detr/test_image_processing_deformable_detr.py +++ b/tests/models/deformable_detr/test_image_processing_deformable_detr.py @@ -162,7 +162,9 @@ def test_image_processor_from_dict_with_kwargs(self): self.assertEqual(image_processor.do_pad, True) image_processor = image_processing_class.from_dict( - self.image_processor_dict, size={"shortest_edge": 42, "longest_edge": 84}, pad_and_return_pixel_mask=False + self.image_processor_dict, + size={"shortest_edge": 42, "longest_edge": 84}, + pad_and_return_pixel_mask=False, ) self.assertEqual(image_processor.size, {"shortest_edge": 42, "longest_edge": 84}) self.assertEqual(image_processor.do_pad, False) diff --git a/tests/models/detr/test_image_processing_detr.py b/tests/models/detr/test_image_processing_detr.py index 5321aa6dbc85..90a927315118 100644 --- a/tests/models/detr/test_image_processing_detr.py +++ b/tests/models/detr/test_image_processing_detr.py @@ -165,7 +165,9 @@ def test_image_processor_from_dict_with_kwargs(self): self.assertEqual(image_processor.do_pad, True) image_processor = image_processing_class.from_dict( - self.image_processor_dict, size={"shortest_edge": 42, "longest_edge": 84}, pad_and_return_pixel_mask=False + self.image_processor_dict, + size={"shortest_edge": 42, "longest_edge": 84}, + pad_and_return_pixel_mask=False, ) self.assertEqual(image_processor.size, {"shortest_edge": 42, "longest_edge": 84}) self.assertEqual(image_processor.do_pad, False)