tool: Auto refresh the lock file on commit

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

.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

pyproject.toml

@@ -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"