diff --git a/yolort/runtime/logits_decoder.py b/yolort/runtime/logits_decoder.py
index a245422d..f47e0cc4 100644
--- 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
index d151caa4..6f4fa86e 100644
--- 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