diff --git a/models/__init__.py b/models/__init__.py
index 7f20684..0acfe73 100644
--- a/models/__init__.py
+++ b/models/__init__.py
@@ -1,9 +1,8 @@
+from .engine import EngineBuilder, TRTModule, TRTProfilerV0, TRTProfilerV1  # isort:skip # noqa: E501
 import warnings
 
 import torch
 
-from .engine import EngineBuilder, TRTModule, TRTProfilerV0, TRTProfilerV1
-
 warnings.filterwarnings(action='ignore', category=torch.jit.TracerWarning)
 warnings.filterwarnings(action='ignore', category=torch.jit.ScriptWarning)
 warnings.filterwarnings(action='ignore', category=UserWarning)