fix lint

open-mmlab · Nov 15, 2022 · a64c6a5 · a64c6a5
1 parent e9e350e
commit a64c6a5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/mmdet3d/models/task_modules/samplers/iou_neg_piecewise_sampler.py b/mmdet3d/models/task_modules/samplers/iou_neg_piecewise_sampler.py
@@ -1,5 +1,6 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 import math
+
 import torch
 
 from mmdet3d.registry import TASK_UTILS
@@ -77,9 +78,9 @@ def _sample_neg(self, assign_result, num_expected, **kwargs):
                     # the same number in the current piece.
                     piece_expected_num = min(
                         num_expected,
-                        math.ceil(
-                            num_expected *
-                            self.neg_piece_fractions[piece_inds]) + extend_num)
+                        math.ceil(num_expected *
+                                  self.neg_piece_fractions[piece_inds]) +
+                        extend_num)
                     min_iou_thr = self.neg_iou_thr[piece_inds + 1]
                 max_iou_thr = self.neg_iou_thr[piece_inds]
                 piece_neg_inds = torch.nonzero(