chore: adding isort to pre-commit

Signed-off-by: Naren Dasan <naren@narendasan.com>
Signed-off-by: Naren Dasan <narens@nvidia.com>

Author: narendasan

.pre-commit-config.yaml

@@ -44,6 +44,7 @@ repos:
     hooks:
     -   id: mypy
         exclude: "^py/torch_tensorrt/fx|^examples|^tests|^tools"
+        python_version: "3.11"
   - repo: local
     hooks:
     -   id: dont-commit-upstream

py/torch_tensorrt/_Device.py

@@ -1,17 +1,18 @@
-from typing import Optional, Any, Tuple
 import sys
+from typing import Any, Optional, Tuple
 
 if sys.version_info >= (3, 11):
     from typing import Self
 else:
     from typing_extensions import Self
 
+import warnings
+
 import torch
+from torch_tensorrt import logging
 
 # from torch_tensorrt import _enums
 import tensorrt as trt
-from torch_tensorrt import logging
-import warnings
 
 try:
     from torch_tensorrt import _C

py/torch_tensorrt/_Input.py

@@ -1,8 +1,7 @@
 from enum import Enum
-from typing import List, Dict, Any, Tuple, Optional, Sequence
+from typing import Any, Dict, List, Optional, Sequence, Tuple
 
 import torch
-
 from torch_tensorrt import _enums
 
 

py/torch_tensorrt/__init__.py

@@ -1,16 +1,16 @@
 import ctypes
 import os
-import sys
 import platform
+import sys
+from typing import Dict, List
+
 from packaging import version
 from torch_tensorrt._version import (
     __cuda_version__,
     __cudnn_version__,
     __tensorrt_version__,
 )
 
-from typing import Dict, List
-
 if sys.version_info < (3,):
     raise Exception(
         "Python 2 has reached end-of-life and is not supported by Torch-TensorRT"
@@ -81,11 +81,11 @@ def _find_lib(name: str, paths: List[str]) -> str:
             ctypes.CDLL(_find_lib(lib, LINUX_PATHS))
 
 import torch
-
 from torch_tensorrt._compile import *  # noqa: F403
-from torch_tensorrt._util import *  # noqa: F403
 from torch_tensorrt._enums import *  # noqa: F403
-
+from torch_tensorrt._util import *  # noqa: F403
+from torch_tensorrt._Input import Input # noqa: F401
+from torch_tensorrt._Device import Device # noqa: F401
 
 if version.parse(sanitized_torch_version()) >= version.parse("2.1.dev"):
     from torch_tensorrt import dynamo  # noqa: F401

py/torch_tensorrt/_compile.py

@@ -1,19 +1,16 @@
-from typing import List, Any, Set, Callable, TypeGuard, Optional
+from enum import Enum
+from typing import Any, Callable, List, Optional, Set, TypeGuard
 
+import torch
+import torch.fx
 import torch_tensorrt.ts
-
 from torch_tensorrt import logging
-from torch_tensorrt._Input import Input
 from torch_tensorrt._enums import dtype
-import torch
-import torch.fx
-from enum import Enum
-
-from torch_tensorrt.fx import InputTensorSpec
-from torch_tensorrt.fx.utils import LowerPrecision
-
+from torch_tensorrt._Input import Input
 from torch_tensorrt.dynamo.compile import compile as dynamo_compile
+from torch_tensorrt.fx import InputTensorSpec
 from torch_tensorrt.fx.lower import compile as fx_compile
+from torch_tensorrt.fx.utils import LowerPrecision
 from torch_tensorrt.ts._compiler import compile as torchscript_compile
 
 

py/torch_tensorrt/_enums.py

@@ -0,0 +1,2 @@
+from torch_tensorrt._C import dtype, EngineCapability, TensorFormat  # noqa: F401
+from tensorrt import DeviceType  # noqa: F401

py/torch_tensorrt/_util.py

@@ -1,7 +1,5 @@
-from torch_tensorrt import __version__
-from torch_tensorrt import _C
-
 import torch
+from torch_tensorrt import _C, __version__
 
 
 def dump_build_info() -> None:

py/torch_tensorrt/dynamo/__init__.py

@@ -4,10 +4,10 @@
 if version.parse(sanitized_torch_version()) >= version.parse("2.1.dev"):
     from ._settings import *  # noqa: F403
     from .conversion import * # noqa: F403
-    from .aten_tracer import trace
+    from .aten_tracer import trace # noqa: F403
     from .conversion.converter_registry import (
-        DYNAMO_CONVERTERS,
-        dynamo_tensorrt_converter,
+        DYNAMO_CONVERTERS, # noqa: F403
+        dynamo_tensorrt_converter, # noqa: F403
     )
-    from .compile import compile
-    from ._SourceIR import SourceIR
+    from .compile import compile # noqa: F403
+    from ._SourceIR import SourceIR # noqa: F403

py/torch_tensorrt/dynamo/_settings.py

@@ -1,17 +1,21 @@
 from dataclasses import dataclass, field
 from typing import Optional, Set
+
 import torch
 from torch_tensorrt.dynamo._defaults import (
-    PRECISION,
     DEBUG,
-    WORKSPACE_SIZE,
-    MIN_BLOCK_SIZE,
-    PASS_THROUGH_BUILD_FAILURES,
     MAX_AUX_STREAMS,
-    VERSION_COMPATIBLE,
+    MIN_BLOCK_SIZE,
     OPTIMIZATION_LEVEL,
+    PASS_THROUGH_BUILD_FAILURES,
+    PRECISION,
     USE_PYTHON_RUNTIME,
+<<<<<<< HEAD
     TRUNCATE_LONG_AND_DOUBLE,
+=======
+    VERSION_COMPATIBLE,
+    WORKSPACE_SIZE,
+>>>>>>> e39abb60d (chore: adding isort to pre-commit)
 )
 
 

py/torch_tensorrt/dynamo/aten_tracer.py

@@ -7,7 +7,6 @@
 import torch._dynamo as torchdynamo
 from torch import _guards
 from torch.fx.passes.infra.pass_base import PassResult
-
 from torch_tensorrt.dynamo.utils import req_torch_version
 from torch_tensorrt.fx.passes.lower_basic_pass_aten import (
     compose_bmm,