Max py (#2017)

Author: mikel-brostrom

.github/workflows/ci.yml

@@ -38,8 +38,8 @@ jobs:
       - name: Install requirements
         shell: bash  # for Windows compatibility
         run: |
-          sed -i'' -e 's/index = "torch-gpu"/index = "torch-cpu"/g' pyproject.toml
           python -m pip install --upgrade pip setuptools wheel uv
+          sed -i'' -e 's/index = "torch-gpu"/index = "torch-cpu"/g' pyproject.toml
           uv sync --group yolo
       - name: Generate detections and embeddings
         run: |
@@ -74,9 +74,9 @@ jobs:
       - name: Install requirements
         shell: bash  # for Windows compatibility
         run: |
-          sed -i'' -e 's/index = "torch-gpu"/index = "torch-cpu"/g' pyproject.toml
           python -m pip install --upgrade pip setuptools wheel uv
-          uv sync --group yolo --group evolve
+          sed -i'' -e 's/index = "torch-gpu"/index = "torch-cpu"/g' pyproject.toml
+          uv sync  --group yolo --group evolve
       - name: Evolve set of parameters for selected tracking method
         run: |
           source .venv/bin/activate
@@ -205,7 +205,7 @@ jobs:
         run: |
           sed -i'' -e 's/index = "torch-gpu"/index = "torch-cpu"/g' pyproject.toml
           python -m pip install --upgrade pip setuptools wheel uv
-          uv sync --group yolo
+          uv sync  --group yolo
       - name: Test tracking with seg models
         env:
           IMG: ./assets/MOT17-mini/train/MOT17-02-FRCNN/img1/000001.jpg
@@ -229,7 +229,7 @@ jobs:
         run: |
           sed -i'' -e 's/index = "torch-gpu"/index = "torch-cpu"/g' pyproject.toml
           python -m pip install --upgrade pip setuptools wheel uv
-          uv sync --all-groups
+          uv sync  --all-groups
           # needed for TFLite export
           sudo apt-get install flatbuffers-compiler
           wget https://github.com/PINTO0309/onnx2tf/releases/download/1.16.31/flatc.tar.gz

pyproject.toml

@@ -7,30 +7,31 @@
 #######################
 
 [[tool.uv.index]]
-name     = "torch-gpu"
-url      = "https://download.pytorch.org/whl/cu121"
+name     = "torch-cpu"
+url      = "https://download.pytorch.org/whl/cpu"
 explicit = true
 
+
 [[tool.uv.index]]
-name     = "torch-cpu"
-url      = "https://download.pytorch.org/whl/cpu"
+name     = "torch-gpu"
+url      = "https://download.pytorch.org/whl/cu121"
 explicit = true
 
-#################################
-# Dependency-to-Index Mapping   #
-#################################
+##################################
+# Pin torch/vision to *an* index #
+##################################
 
+[tool.uv.sources]
 torch = [
-  # CUDA build on Linux/Windows only
-  { index = "torch-gpu", marker = "sys_platform != 'darwin'" },
-  # CPU build on macOS
-  { index = "torch-cpu", marker = "sys_platform == 'darwin'" }
+  { index = "torch-cpu", marker = "sys_platform == 'darwin'" },
+  { index = "torch-gpu", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
 ]
 torchvision = [
-  { index = "torch-gpu", marker = "sys_platform != 'darwin'" },
-  { index = "torch-cpu", marker = "sys_platform == 'darwin'" }
+  { index = "torch-cpu", marker = "sys_platform == 'darwin'" },
+  { index = "torch-gpu", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
 ]
 
+
 #########################
 # UV-specific Overrides #
 #########################
@@ -43,6 +44,7 @@ requires-dist = ["onnx>=1.17.0", "onnxsim<1.0.0,>=0.4.36"]
 [tool.uv]
 no-build-isolation-package = ["yolox"]
 
+
 ############
 # Flake8   #
 ############
@@ -111,12 +113,12 @@ keywords = [
 ]
 
 dependencies = [
-  "setuptools>=78.0.0",
+  "setuptools",
   "filterpy<2.0.0,>=1.4.5",
   "gdown<6.0.0,>=5.1.0",
   "lapx<1.0.0,>=0.5.5",
   "loguru<1.0.0,>=0.7.2",
-  "numpy==1.26.4",
+  "numpy",
   "pyyaml<7.0.0,>=6.0.1",
   "regex<2025.0.0,>=2024.0.0",
   "yacs<1.0.0,>=0.1.8",
@@ -155,7 +157,7 @@ evolve = [
 dev = ["ipykernel<7.0.0,>=6.29.5"]
 
 yolox-build-deps = [
-  "setuptools>=78.0.0",
+  "setuptools",
   "torch<3.0.0,>=2.2.1",
 ]