Apply pre-commit

zhiqwang · Jan 24, 2022 · a1f9059 · a1f9059
1 parent 88a4c22
commit a1f9059
Showing 1 changed file with 7 additions and 7 deletions.
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