From 85844a3a9e4e3c20ec1ad72b58a9a88202a41c2f Mon Sep 17 00:00:00 2001
From: lixiaojie <lixiaojie@sensetime.com>
Date: Sun, 14 Jun 2020 01:25:08 +0800
Subject: [PATCH] fix linting

---
 tests/test_backbones/test_shufflenet_v1.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_backbones/test_shufflenet_v1.py b/tests/test_backbones/test_shufflenet_v1.py
index 57c6105c7..59d190891 100644
--- a/tests/test_backbones/test_shufflenet_v1.py
+++ b/tests/test_backbones/test_shufflenet_v1.py
@@ -221,14 +221,14 @@ def test_shufflenetv1_backbone():
         if is_norm(m):
             assert isinstance(m, _BatchNorm)
 
-    imgs = torch.randn(1, 3, 224,  224)
+    imgs = torch.randn(1, 3, 224, 224)
     feat = model(imgs)
     assert len(feat) == 2
     assert feat[0].shape == torch.Size((1, 480, 14, 14))
     assert feat[1].shape == torch.Size((1, 960, 7, 7))
 
     # Test ShuffleNetv1 forward with layers 2 forward
-    model = ShuffleNetv1(groups=3, out_indices=(2,))
+    model = ShuffleNetv1(groups=3, out_indices=(2, ))
     model.init_weights()
     model.train()