tool: Auto refresh the lock file on commit

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

Author: narendasan

.pre-commit-config.yaml

@@ -55,6 +55,12 @@ repos:
     rev: v1.22.9
     hooks:
       - id: typos
+  - repo: https://github.com/astral-sh/uv-pre-commit
+    # uv version.
+    rev: 0.4.10
+    hooks:
+      # Update the uv lockfile
+      - id: uv-lock
   - repo: local
     hooks:
     -   id: dont-commit-upstream

cpp/include/torch_tensorrt/macros.h

@@ -24,7 +24,7 @@
 #define STR(x) XSTR(x)
 
 #define TORCH_TENSORRT_MAJOR_VERSION 2
-#define TORCH_TENSORRT_MINOR_VERSION 6
+#define TORCH_TENSORRT_MINOR_VERSION 5
 #define TORCH_TENSORRT_PATCH_VERSION 0
 #define TORCH_TENSORRT_VERSION      \
   STR(TORCH_TENSORRT_MAJOR_VERSION) \

pyproject.toml

@@ -9,7 +9,7 @@ requires = [
     "typing-extensions>=4.7.0",
     "future>=0.18.3",
     "tensorrt-cu12==10.3.0",
-    "torch>=2.6.0.dev,<2.7.0",
+    "torch>=2.5.0.dev,<2.6.0",
     "pybind11==2.6.2",
     "numpy",
 ]
@@ -54,7 +54,7 @@ keywords = [
     "inference",
 ]
 dependencies = [
-    "torch>=2.6.0.dev,<2.7.0",
+    "torch>=2.5.0.dev,<2.6.0",
     "tensorrt-cu12==10.3.0",
     "tensorrt-cu12-bindings==10.3.0",
     "tensorrt-cu12-libs==10.3.0",
@@ -65,7 +65,9 @@ dependencies = [
 dynamic = ["version"]
 
 [project.optional-dependencies]
-#torchvision = ["torchvision>=0.21.dev,<0.22.0"] # Reenable once torchvision bumps to 2.6.0
+torchvision = [
+    "torchvision",
+] #Leaving torchvisions dependency unconstrained so uv can just install something that should work for the torch we have. TV's on PyT makes it hard to put version constrains in
 quantization = ["nvidia-modelopt[deploy,hf,torch]~=0.17.0"]
 monitoring-tools = ["rich>=13.7.1"]
 jupyter = ["rich[jupyter]>=13.7.1"]
@@ -85,22 +87,21 @@ dev-dependencies = [
     "pre-commit>=2.20.0",
     "black>=22.6.0",
     "clang-format==14.0.6",
+    "typos",
     "mypy",
     "isort",
     "ruff",
     "pytest",
-    "parameterized",
+    "pytest-xdist",
+    "parameterized>=0.2.0",
     "expecttest==0.1.6",
     "pyyaml",
 ]
 
-environments = [
-    "sys_platform == 'linux'",
-    "sys_platform == 'windows'",
-]
+environments = ["sys_platform == 'linux'", "sys_platform == 'windows'"]
 
 extra-index-url = [
-    "https://download.pytorch.org/whl/nightly/cu124" # We are going to define the dev enviorment as latest supported CUDA, and allow CI to handle the others, change as needed
+    "https://download.pytorch.org/whl/nightly/cu124", # We are going to define the dev enviorment as latest supported CUDA, and allow CI to handle the others, change as needed
 ]
 
 prerelease = "if-necessary-or-explicit"