delete cpu dependecies

mikel-brostrom · May 12, 2024 · 5e48c74 · 5e48c74
1 parent fbf11c6
commit 5e48c74
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -41,7 +41,7 @@ jobs:
           python -m pip install --upgrade pip setuptools wheel
           pip install poetry
           poetry config virtualenvs.create false
-          poetry install --with cpu,yolo
+          poetry install --with yolo
 
       - name: Generate detections and embeddings
         run: |
@@ -80,7 +80,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with cpu,yolo,evolve
+          poetry install --with yolo,evolve
 
       - name: Generate detections and embeddings
         run: |
@@ -116,7 +116,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with cpu
+          poetry install
 
       - name: Generate detections and embeddings
         run: |
@@ -141,7 +141,7 @@ jobs:
       - run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with cpu,yolo
+          poetry install --with yolo
       - name: Test tracking with pose models
         env:
           IMG: ./assets/MOT17-mini/train/MOT17-02-FRCNN/img1/000001.jpg
@@ -160,7 +160,7 @@ jobs:
       - run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with cpu,yolo
+          poetry install --with yolo
       - name: Test tracking with pose models
         env:
           IMG: ./assets/MOT17-mini/train/MOT17-02-FRCNN/img1/000001.jpg
@@ -186,7 +186,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with cpu,yolo
+          poetry install --with yolo
       - name: Test tracking with seg models
         env:
           IMG: ./assets/MOT17-mini/train/MOT17-02-FRCNN/img1/000001.jpg
@@ -209,7 +209,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with cpu,export
+          poetry install --with export
       - name: Test export models
         run: python boxmot/appearance/reid_export.py --include torchscript onnx openvino --device cpu --batch-size 3 --dynamic
       - name: Test inference on exported models
@@ -234,7 +234,7 @@ jobs:
       - run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with cpu,test
+          poetry install --with test
       - name: Test export models
         run: python boxmot/appearance/reid_export.py --include torchscript onnx openvino --device cpu --batch-size 3 --dynamic
       - name: Pytest tests  # after tracking options as this does not download models

diff --git a/pyproject.toml b/pyproject.toml
@@ -51,6 +51,8 @@ torchvision = [
     {version = "^0.17.1", source="torch_cuda121", markers = "sys_platform == 'win32'"},
 ]
 
+
+
 [tool.poetry.group.test.dependencies]
 pytest = "^8.0.2"
 isort = "^5.13.2"