Merge pull request #2126 from pytorch/mypy

Improve Python tooling

Author: narendasan

.pre-commit-config.yaml

@@ -1,10 +1,11 @@
 exclude: ^.github/actions/assigner/dist
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v2.3.0
+    rev: v4.4.0
     hooks:
       - id: check-yaml
       - id: trailing-whitespace
+        exclude: ^docs
       - id: check-added-large-files
         args:
           - --maxkb=1000
@@ -13,18 +14,14 @@ repos:
       - id: mixed-line-ending
         args:
           - --fix=lf
-  - repo: https://github.com/psf/black
-    rev: 22.6.0
-    hooks:
-      - id: black
-        exclude: ^examples/custom_converters/elu_converter/setup.py
+        exclude: ^docs
   - repo: https://github.com/pre-commit/mirrors-clang-format
-    rev: v14.0.6
+    rev: v16.0.6
     hooks:
       - id: clang-format
         types_or: [c++, c, cuda]
   - repo: https://github.com/keith/pre-commit-buildifier
-    rev: 5.1.0
+    rev: 6.1.0.2
     hooks:
     -   id: buildifier
         args:
@@ -34,6 +31,26 @@ repos:
     rev: v0.13
     hooks:
       - id: validate-pyproject
+  - repo: https://github.com/pycqa/isort
+    rev: 5.12.0
+    hooks:
+      - id: isort
+        name: isort (python)
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: 'v1.4.1'
+    hooks:
+    -   id: mypy
+        exclude: "^py/torch_tensorrt/fx|^examples|^tests|^tools|^docs|noxfile.py|setup.py|versions.py"
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    # Ruff version.
+    rev: v0.0.278
+    hooks:
+      - id: ruff
+  - repo: https://github.com/psf/black
+    rev: 23.7.0
+    hooks:
+      - id: black
+        exclude: ^examples/custom_converters/elu_converter/setup.py|^docs
   - repo: local
     hooks:
     -   id: dont-commit-upstream
@@ -42,4 +59,3 @@ repos:
         exclude: "^.pre-commit-config.yaml"
         language: pygrep
         types: [text]
-

BUILD

@@ -47,8 +47,8 @@ pkg_tar(
         ":windows": ["//cpp/lib:torch_tensorrt.dll"],
         "//conditions:default": [
             "//cpp/lib:libtorchtrt.so",
-            "//cpp/lib:libtorchtrt_runtime.so",
             "//cpp/lib:libtorchtrt_plugins.so",
+            "//cpp/lib:libtorchtrt_runtime.so",
         ],
     }),
     mode = "0755",
@@ -74,9 +74,9 @@ pkg_tar(
     extension = "tar.gz",
     package_dir = "torch_tensorrt",
     deps = [
-        ":lib",
         ":include",
         ":include_core",
+        ":lib",
     ] + select({
         ":windows": [],
         "//conditions:default": [":bin"],