From 0d473ccd4e8fcbe774428dc32a1bfd17814e4620 Mon Sep 17 00:00:00 2001
From: huyingfan <huyingfan@sensetime.com>
Date: Wed, 23 Mar 2022 20:41:10 +0800
Subject: [PATCH] lint

---
 mmcls/core/utils/__init__.py          |  4 +++-
 mmcls/core/utils/dist_utils.py        |  2 +-
 mmcls/datasets/builder.py             |  3 ++-
 mmcls/datasets/samplers/repeat_aug.py | 20 +++++++++-----------
 4 files changed, 15 insertions(+), 14 deletions(-)

diff --git a/mmcls/core/utils/__init__.py b/mmcls/core/utils/__init__.py
index 51a575d0901..7170f232d3f 100644
--- a/mmcls/core/utils/__init__.py
+++ b/mmcls/core/utils/__init__.py
@@ -2,4 +2,6 @@
 from .dist_utils import DistOptimizerHook, allreduce_grads, sync_random_seed
 from .misc import multi_apply
 
-__all__ = ['allreduce_grads', 'DistOptimizerHook', 'multi_apply', 'sync_random_seed']
+__all__ = [
+    'allreduce_grads', 'DistOptimizerHook', 'multi_apply', 'sync_random_seed'
+]
diff --git a/mmcls/core/utils/dist_utils.py b/mmcls/core/utils/dist_utils.py
index 30cafdbd15a..8912cea435a 100644
--- a/mmcls/core/utils/dist_utils.py
+++ b/mmcls/core/utils/dist_utils.py
@@ -95,4 +95,4 @@ def sync_random_seed(seed=None, device='cuda'):
     else:
         random_num = torch.tensor(0, dtype=torch.int32, device=device)
     dist.broadcast(random_num, src=0)
-    return random_num.item()
\ No newline at end of file
+    return random_num.item()
diff --git a/mmcls/datasets/builder.py b/mmcls/datasets/builder.py
index f73c32fca3f..1c74af9c42b 100644
--- a/mmcls/datasets/builder.py
+++ b/mmcls/datasets/builder.py
@@ -104,7 +104,8 @@ def build_dataloader(dataset,
         sampler = build_sampler(
             sampler_cfg,
             default_args=dict(
-                dataset=dataset, num_replicas=world_size, rank=rank, seed=seed))
+                dataset=dataset, num_replicas=world_size, rank=rank,
+                seed=seed))
     # Default sampler logic
     elif dist:
         sampler = build_sampler(
diff --git a/mmcls/datasets/samplers/repeat_aug.py b/mmcls/datasets/samplers/repeat_aug.py
index e1863857c79..5de096bdef2 100644
--- a/mmcls/datasets/samplers/repeat_aug.py
+++ b/mmcls/datasets/samplers/repeat_aug.py
@@ -21,17 +21,15 @@ class RepeatAugSampler(Sampler):
     Copyright (c) 2015-present, Facebook, Inc.
     """
 
-    def __init__(
-        self,
-        dataset,
-        num_replicas=None,
-        rank=None,
-        shuffle=True,
-        num_repeats=3,
-        selected_round=256,
-        selected_ratio=0,
-        seed=0
-    ):
+    def __init__(self,
+                 dataset,
+                 num_replicas=None,
+                 rank=None,
+                 shuffle=True,
+                 num_repeats=3,
+                 selected_round=256,
+                 selected_ratio=0,
+                 seed=0):
         default_rank, default_world_size = get_dist_info()
         rank = default_rank if rank is None else rank
         num_replicas = (