feat: add Python setup and ONNX model generation to CI workflow

2026-05-23 13:16:08 +00:00
parent 5b4f8d29df
commit 8d04a17d84
2 changed files with 21 additions and 6 deletions
@@ -37,6 +37,20 @@ jobs:
      - name: Set up Docker Buildx
        uses: docker/setup-buildx-action@v3

+      - name: Set up Python for ONNX conversion
+        if: matrix.service.name == 'ml'
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'
+
+      - name: Generate ONNX model artifact
+        if: matrix.service.name == 'ml'
+        working-directory: Machine_Learning
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install 'tensorflow>=2.13.0' 'tf2onnx>=1.16.1' 'onnx>=1.16.0'
+          python convert_onnx.py
+
      - name: Log in to GHCR
        uses: docker/login-action@v3
        with: