sahi/.github/workflows/ci.yml at 61488bde47f3b8e4936956a7d56d488e29d9d527 · obss/sahi · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
name: "CI"
on:
  push:
    branches: [main]
    paths-ignore:
      - "**.md"
      - "**.ipynb"
      - "**.cff"
      - "docs/**"
      - "demo/**"
      - "context7.json"
      - "mkdocs.yml"
      - "zensical.toml"

  pull_request:
    branches: [main]
    paths-ignore:
      - "**.md"
      - "**.ipynb"
      - "**.cff"
      - "docs/**"
      - "demo/**"
      - "context7.json"
      - "mkdocs.yml"
      - "zensical.toml"

  schedule:
    - cron: "0 0 * * *" # Runs at 00:00 UTC every day

  workflow_dispatch: # allow running sync via github ui button

jobs:
  ci:
    name: "Tests (OS: ${{ matrix.os }}, Python: ${{ matrix.python-version }})"
    if:
      github.event_name != 'schedule' && github.event_name !=
      'workflow_dispatch'
    runs-on: ${{ matrix.os }}
    strategy:
      fail-fast: false
      matrix:
        os: [ubuntu-latest]
        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
        include:
          - os: macos-latest
            python-version: "3.12"
          - os: windows-latest
            python-version: "3.12"
    steps:
      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
      - name: Setup uv python package manager
        uses: astral-sh/setup-uv@cec208311dfd045dd5311c1add060b2062131d57 # v8.0.0
        with:
          python-version: ${{ matrix.python-version }}
          enable-cache: true
          prune-cache: false
          activate-environment: true
          cache-dependency-glob: |
            **/pyproject.toml

      - name: Cache model weights
        uses: actions/cache@v5
        with:
          path: tests/data/models
          key:
            ${{ runner.os }}-models-${{ hashFiles('sahi/utils/huggingface.py',
            'sahi/utils/yolov5.py', 'sahi/utils/ultralytics.py',
            'sahi/utils/rtdetr.py', 'sahi/utils/torchvision.py') }}
          restore-keys: |
            ${{ runner.os }}-models-

      - name: Install dependencies
        run: uv sync --group dev --extra ci
      - name: List installed packages
        run: |
          echo "=== Installed Python packages ==="
          uv pip list
          echo ""
          echo "=== Python version ==="
          python --version
          echo ""
          echo "=== uv version ==="
          uv --version

      - name: Test with python ${{ matrix.python-version }}
        run: |
          # Exclude MMDet tests which run in a separate workflow
          uv run pytest --capture=no -n auto -k "not (mmdet or test_get_prediction_mmdet or test_get_sliced_prediction_mmdet or test_mmdet_yolox_tiny_prediction)"
      - name: Test SAHI CLI (without MMDet)
        run: |
          uv run sahi predict --no_sliced_prediction --model_type ultralytics --source tests/data/coco_utils/terrain1.jpg --novisual --model_path tests/data/models/ultralytics/yolo11n.pt --image_size 320
          uv run sahi predict --model_type ultralytics --source tests/data/ --novisual --model_path tests/data/models/ultralytics/yolo11n.pt --image_size 320
          uv run sahi predict --model_type ultralytics --source tests/data/coco_utils/terrain1.jpg --export_pickle --export_crop --model_path tests/data/models/ultralytics/yolo11n.pt --image_size 320
          uv run sahi predict --model_type ultralytics --source tests/data/coco_utils/ --novisual --dataset_json_path tests/data/coco_utils/combined_coco.json --model_path tests/data/models/ultralytics/yolo11n.pt --image_size 320
          # coco yolov5
          uv run sahi coco yolov5 --image_dir tests/data/coco_utils/ --dataset_json_path tests/data/coco_utils/combined_coco.json --train_split 0.9
          # coco evaluate
          uv run sahi coco evaluate --dataset_json_path tests/data/coco_evaluate/dataset.json --result_json_path tests/data/coco_evaluate/result.json
          # coco analyse
          uv run sahi coco analyse --dataset_json_path tests/data/coco_evaluate/dataset.json --result_json_path tests/data/coco_evaluate/result.json --out_dir tests/data/coco_evaluate/

  schedule-test:
    name:
      "Schedule/PyPI check (OS: ${{ matrix.os }}, Python: ${{
      matrix.python-version }})"
    if:
      github.event_name == 'schedule' || github.event_name ==
      'workflow_dispatch'
    runs-on: ${{ matrix.os }}
    strategy:
      fail-fast: false
      matrix:
        os: [ubuntu-latest]
        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
        include:
          - os: macos-latest
            python-version: "3.12"
          - os: windows-latest
            python-version: "3.12"
    steps:
      - name: Setup uv python package manager
        uses: astral-sh/setup-uv@cec208311dfd045dd5311c1add060b2062131d57 # v8.0.0
        with:
          python-version: ${{ matrix.python-version }}
          enable-cache: true
          prune-cache: false
          activate-environment: true
      - name: List installed packages
        run: |
          echo "=== Python version ==="
          python --version
          echo ""
          echo "=== uv version ==="
          uv --version

      - name: Install sahi from PyPI
        run: |
          uv pip install sahi
          uv pip show sahi
          uv run python -c "import sahi; print(sahi.__version__)"