chore: adding isort to pre-commit

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

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,9 @@ 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
 
 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/_settings.py

@@ -1,16 +1,17 @@
 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,
+    VERSION_COMPATIBLE,
+    WORKSPACE_SIZE,
 )
 
 

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,

py/torch_tensorrt/dynamo/backend/backends.py

@@ -1,25 +1,16 @@
 import logging
-from typing import Sequence, Any, Callable
-import torch
 from functools import partial
-import torch._dynamo as td
+from typing import Any, Callable, Sequence
 
+import torch
+import torch._dynamo as td
+from torch._functorch.aot_autograd import aot_module_simplified, make_boxed_compiler
 from torch_tensorrt.dynamo import CompilationSettings
-from torch_tensorrt.dynamo.lowering._decompositions import (
-    get_decompositions,
-)
-from torch_tensorrt.dynamo.lowering._pre_aot_lowering import (
-    pre_aot_substitutions,
-)
-from torch_tensorrt.dynamo.lowering._partition import (
-    partition,
-    get_submod_inputs,
-)
-from torch_tensorrt.dynamo.utils import parse_dynamo_kwargs
 from torch_tensorrt.dynamo.conversion import convert_module
-
-from torch._functorch.aot_autograd import aot_module_simplified, make_boxed_compiler
-
+from torch_tensorrt.dynamo.lowering._decompositions import get_decompositions
+from torch_tensorrt.dynamo.lowering._partition import get_submod_inputs, partition
+from torch_tensorrt.dynamo.lowering._pre_aot_lowering import pre_aot_substitutions
+from torch_tensorrt.dynamo.utils import parse_dynamo_kwargs
 
 logger = logging.getLogger(__name__)