Reformatting

HichTala · Nov 28, 2024 · c3040ac · c3040ac
1 parent ac4522e
commit c3040ac
Show file tree

Hide file tree

Showing 10 changed files with 30 additions and 10 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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)