Apply pre-commit

zhiqwang · Jan 24, 2022 · c72dd6a · c72dd6a
1 parent 88a4c22
commit c72dd6a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/yolort/runtime/logits_decoder.py b/yolort/runtime/logits_decoder.py
@@ -1,5 +1,6 @@
 # Copyright (c) 2021, yolort team. All rights reserved.
 from typing import List, Tuple
+
 import torch
 from torch import nn, Tensor
 from yolort.models.box_head import _concat_pred_logits, _decode_pred_logits

diff --git a/yolort/runtime/yolo_graphsurgeon.py b/yolort/runtime/yolo_graphsurgeon.py
@@ -156,13 +156,13 @@ def register_nms(
 
         op = "EfficientNMS_TRT"
         attrs = {
-            'plugin_version': "1",
-            'background_class': -1,  # no background class
-            'max_output_boxes': detections_per_img,
-            'score_threshold': score_thresh,
-            'iou_threshold': nms_thresh,
-            'score_activation': False,
-            'box_coding': 0,
+            "plugin_version": "1",
+            "background_class": -1,  # no background class
+            "max_output_boxes": detections_per_img,
+            "score_threshold": score_thresh,
+            "iou_threshold": nms_thresh,
+            "score_activation": False,
+            "box_coding": 0,
         }
 
         # NMS Outputs