diff --git a/scripts/infer.py b/scripts/infer.py
index 856a7c0..313a7a2 100644
--- a/scripts/infer.py
+++ b/scripts/infer.py
@@ -33,9 +33,7 @@
 from .train import CONFIG
 from .utils.trans_utils import VistaPostTransform, get_largest_connected_component_point
 
-trt_wrap, TRT_AVAILABLE = optional_import(
-    "monai.networks", name="trt_wrap"
-)
+trt_wrap, TRT_AVAILABLE = optional_import("monai.networks", name="trt_wrap")
 
 rearrange, _ = optional_import("einops", name="rearrange")
 sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
@@ -143,7 +141,7 @@ def __init__(self, config_file="./configs/infer.yaml", **override):
                     "builder_optimization_level": 5,
                     "precision_constraints": "obey",
                 },
-                "timestamp": ts
+                "timestamp": ts,
             }
 
             trt_wrap(