fix lint

Signed-off-by: wep21 <[email protected]>

Author: wep21

projects/TransFusion/transfusion/transfusion_head.py

@@ -194,9 +194,8 @@ def forward_single(self, inputs, metas):
         #################################
         # image to BEV
         #################################
-        lidar_feat_flatten = lidar_feat.view(batch_size,
-                                               lidar_feat.shape[1],
-                                               -1)  # [BS, C, H*W]
+        lidar_feat_flatten = lidar_feat.view(batch_size, lidar_feat.shape[1],
+                                             -1)  # [BS, C, H*W]
         bev_pos = self.bev_pos.repeat(batch_size, 1, 1).to(lidar_feat.device)
 
         #################################
@@ -702,7 +701,8 @@ def get_targets_single(self, gt_instances_3d, preds_dict, batch_idx):
                                       device=device)
                 center_int = center.to(torch.int32)
 
-                draw_heatmap_gaussian(heatmap[gt_labels_3d[idx]], center_int, radius)
+                draw_heatmap_gaussian(heatmap[gt_labels_3d[idx]], center_int,
+                                      radius)
 
         mean_iou = ious[pos_inds].sum() / max(len(pos_inds), 1)
         return (
@@ -843,4 +843,4 @@ def loss_by_feat(self, preds_dicts: Tuple[List[dict]],
 
         loss_dict['matched_ious'] = layer_loss_cls.new_tensor(matched_ious)
 
-        return loss_dict
+        return loss_dict