Lint using yapf=0.32.0 on python 3.10.19

Author: MGAMZ

mmengine/_strategy/deepspeed.py

@@ -310,10 +310,10 @@ def __init__(
             self.config.setdefault('gradient_accumulation_steps', 1)
         self.config['steps_per_print'] = steps_per_print
         self._inputs_to_half = inputs_to_half
-        assert (exclude_frozen_parameters is None or digit_version(
-            deepspeed.__version__) >= digit_version('0.13.2')), (
-                'DeepSpeed >= 0.13.2 is required to enable '
-                'exclude_frozen_parameters')
+        assert (exclude_frozen_parameters is None or
+                digit_version(deepspeed.__version__) >= digit_version('0.13.2')
+                ), ('DeepSpeed >= 0.13.2 is required to enable '
+                    'exclude_frozen_parameters')
         self.exclude_frozen_parameters = exclude_frozen_parameters
 
         register_deepspeed_optimizers()

mmengine/config/config.py

@@ -46,10 +46,9 @@
 def _lazy2string(cfg_dict, dict_type=None):
     if isinstance(cfg_dict, dict):
         dict_type = dict_type or type(cfg_dict)
-        return dict_type({
-            k: _lazy2string(v, dict_type)
-            for k, v in dict.items(cfg_dict)
-        })
+        return dict_type(
+            {k: _lazy2string(v, dict_type)
+             for k, v in dict.items(cfg_dict)})
     elif isinstance(cfg_dict, (tuple, list)):
         return type(cfg_dict)(_lazy2string(v, dict_type) for v in cfg_dict)
     elif isinstance(cfg_dict, (LazyAttr, LazyObject)):
@@ -272,15 +271,13 @@ def __reduce_ex__(self, proto):
         # called by CPython interpreter during pickling. See more details in
         # https://github.com/python/cpython/blob/8d61a71f9c81619e34d4a30b625922ebc83c561b/Objects/typeobject.c#L6196  # noqa: E501
         if digit_version(platform.python_version()) < digit_version('3.8'):
-            return (self.__class__, ({
-                k: v
-                for k, v in super().items()
-            }, ), None, None, None)
+            return (self.__class__, ({k: v
+                                      for k, v in super().items()}, ), None,
+                    None, None)
         else:
-            return (self.__class__, ({
-                k: v
-                for k, v in super().items()
-            }, ), None, None, None, None)
+            return (self.__class__, ({k: v
+                                      for k, v in super().items()}, ), None,
+                    None, None, None)
 
     def __eq__(self, other):
         if isinstance(other, ConfigDict):

mmengine/dataset/utils.py

@@ -158,8 +158,7 @@ def default_collate(data_batch: Sequence) -> Any:
                 return [default_collate(samples) for samples in transposed]
     elif isinstance(data_item, Mapping):
         return data_item_type({
-            key:
-            default_collate([d[key] for d in data_batch])
+            key: default_collate([d[key] for d in data_batch])
             for key in data_item
         })
     else:

mmengine/fileio/backends/local_backend.py

@@ -156,8 +156,8 @@ def isfile(self, filepath: Union[str, Path]) -> bool:
         """
         return osp.isfile(filepath)
 
-    def join_path(self, filepath: Union[str, Path], *filepaths:
-                  Union[str, Path]) -> str:
+    def join_path(self, filepath: Union[str, Path],
+                  *filepaths: Union[str, Path]) -> str:
         r"""Concatenate all file paths.
 
         Join one or more filepath components intelligently. The return value

mmengine/fileio/file_client.py

@@ -385,8 +385,8 @@ def isfile(self, filepath: Union[str, Path]) -> bool:
         """
         return self.client.isfile(filepath)
 
-    def join_path(self, filepath: Union[str, Path], *filepaths:
-                  Union[str, Path]) -> str:
+    def join_path(self, filepath: Union[str, Path],
+                  *filepaths: Union[str, Path]) -> str:
         r"""Concatenate all file paths.
 
         Join one or more filepath components intelligently. The return value

mmengine/hooks/checkpoint_hook.py

@@ -196,10 +196,10 @@ def __init__(self,
         self.save_best = save_best
 
         # rule logic
-        assert (isinstance(rule, str) or is_list_of(rule, str) or
-                (rule
-                 is None)), ('"rule" should be a str or list of str or None, '
-                             f'but got {type(rule)}')
+        assert (isinstance(rule, str) or is_list_of(rule, str)
+                or (rule is None)), (
+                    '"rule" should be a str or list of str or None, '
+                    f'but got {type(rule)}')
         if isinstance(rule, list):
             # check the length of rule list
             assert len(rule) in [

mmengine/model/test_time_aug.py

@@ -124,10 +124,9 @@ def test_step(self, data):
         data_list: Union[List[dict], List[list]]
         if isinstance(data, dict):
             num_augs = len(data[next(iter(data))])
-            data_list = [{
-                key: value[idx]
-                for key, value in data.items()
-            } for idx in range(num_augs)]
+            data_list = [{key: value[idx]
+                          for key, value in data.items()}
+                         for idx in range(num_augs)]
         elif isinstance(data, (tuple, list)):
             num_augs = len(data[0])
             data_list = [[_data[idx] for _data in data]

mmengine/runner/checkpoint.py

@@ -601,10 +601,9 @@ def _load_checkpoint_to_model(model,
     # strip prefix of state_dict
     metadata = getattr(state_dict, '_metadata', OrderedDict())
     for p, r in revise_keys:
-        state_dict = OrderedDict({
-            re.sub(p, r, k): v
-            for k, v in state_dict.items()
-        })
+        state_dict = OrderedDict(
+            {re.sub(p, r, k): v
+             for k, v in state_dict.items()})
     # Keep metadata in state_dict
     state_dict._metadata = metadata
 

mmengine/utils/dl_utils/torch_ops.py

@@ -4,9 +4,9 @@
 from ..version_utils import digit_version
 from .parrots_wrapper import TORCH_VERSION
 
-_torch_version_meshgrid_indexing = ('parrots' not in TORCH_VERSION
-                                    and digit_version(TORCH_VERSION)
-                                    >= digit_version('1.10.0a0'))
+_torch_version_meshgrid_indexing = (
+    'parrots' not in TORCH_VERSION
+    and digit_version(TORCH_VERSION) >= digit_version('1.10.0a0'))
 
 
 def torch_meshgrid(*tensors):

mmengine/visualization/visualizer.py

@@ -754,9 +754,8 @@ def draw_bboxes(
         assert bboxes.shape[-1] == 4, (
             f'The shape of `bboxes` should be (N, 4), but got {bboxes.shape}')
 
-        assert (bboxes[:, 0] <= bboxes[:, 2]).all() and (bboxes[:, 1]
-                                                         <= bboxes[:,
-                                                                   3]).all()
+        assert (bboxes[:, 0] <= bboxes[:, 2]).all() and (bboxes[:, 1] <=
+                                                         bboxes[:, 3]).all()
         if not self._is_posion_valid(bboxes.reshape((-1, 2, 2))):
             warnings.warn(
                 'Warning: The bbox is out of bounds,'