complexity-framework/.github/workflows/detector-export.yml at main · Complexity-ML/complexity-framework · 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
name: Detector export

on:
  pull_request:
    paths:
      - ".github/workflows/detector-export.yml"
      - "complexity/deploy/onnx_detector/**"
      - "complexity/generative/detection/**"
      - "scripts/build_onnx_release.py"
      - "scripts/check_onnx_parity.py"
      - "scripts/onnx_detect.py"
      - "scripts/export_onnx.py"
      - "scripts/export_tensorrt.py"
      - "docs/onnx/release.json"
      - "tests/test_detector_export.py"
      - "tests/test_onnx_release.py"
      - "tests/test_onnx_detect_cli.py"
      - "tests/test_onnx_detector_*.py"
      - "pyproject.toml"
  push:
    branches: [main]
    paths:
      - ".github/workflows/detector-export.yml"
      - "complexity/deploy/onnx_detector/**"
      - "complexity/generative/detection/**"
      - "scripts/build_onnx_release.py"
      - "scripts/check_onnx_parity.py"
      - "scripts/onnx_detect.py"
      - "scripts/export_onnx.py"
      - "scripts/export_tensorrt.py"
      - "docs/onnx/release.json"
      - "tests/test_detector_export.py"
      - "tests/test_onnx_release.py"
      - "tests/test_onnx_detect_cli.py"
      - "tests/test_onnx_detector_*.py"
      - "pyproject.toml"

permissions:
  contents: read

jobs:
  onnx-parity:
    runs-on: ubuntu-latest
    timeout-minutes: 20
    steps:
      - uses: actions/checkout@v7
      - uses: actions/setup-python@v7
        with:
          python-version: "3.11"
          cache: pip
      - name: Install CPU test environment
        run: |
          python -m pip install --upgrade pip
          python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
          python -m pip install -e ".[dev,export]"
      - name: Lint detector export code
        run: >-
          ruff check
          complexity/deploy/onnx_detector
          complexity/generative/detection/exporting.py
          scripts/build_onnx_release.py
          scripts/check_onnx_parity.py
          scripts/onnx_detect.py
          scripts/export_onnx.py
          scripts/export_tensorrt.py
          tests/test_detector_export.py
          tests/test_onnx_detect_cli.py
          tests/test_onnx_detector_core.py
          tests/test_onnx_detector_metadata.py
          tests/test_onnx_detector_pipeline.py
          tests/test_onnx_detector_skeleton.py
          tests/test_onnx_release.py
      - name: Test ONNX branches and dynamic batch parity
        run: >-
          pytest -q
          tests/test_detector_export.py
          tests/test_onnx_detect_cli.py
          tests/test_onnx_detector_core.py
          tests/test_onnx_detector_metadata.py
          tests/test_onnx_detector_pipeline.py
          tests/test_onnx_detector_skeleton.py
          tests/test_onnx_release.py