CI/CD for ML on Kubernetes

Train → Track → Package → Test → Deploy → Monitor
  ↑                                          |
  └──────────── Retrain on drift ←──────────┘

name: ML Model Deploy
on:
  push:
    paths:
      - 'models/**'
      - 'training/**'

jobs:
  train-and-deploy:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v4

      - name: Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '3.11'

      - name: Install dependencies
        run: pip install mlflow scikit-learn

      - name: Train model
        run: python training/train.py
        env:
          MLFLOW_TRACKING_URI: ${{ secrets.MLFLOW_URI }}

      - name: Build Docker image
        run: |
          mlflow models build-docker \
            -m "models:/wine-quality/Production" \
            -n "wine-quality:${{ github.sha }}"

      - name: Push to registry
        run: |
          docker tag wine-quality:${{ github.sha }} \
            ${{ secrets.REGISTRY }}/wine-quality:${{ github.sha }}
          docker push ${{ secrets.REGISTRY }}/wine-quality:${{ github.sha }}

      - name: Deploy to Kubernetes
        run: |
          kubectl set image deployment/wine-quality \
            wine-quality=${{ secrets.REGISTRY }}/wine-quality:${{ github.sha }}

      - name: Test model
        run: |
          docker run -d -p 8080:8080 wine-quality:${{ github.sha }}
          sleep 10
          curl -f http://localhost:8080/health
          python tests/test_inference.py

# Quick rollback
kubectl rollout undo deployment/wine-quality

# Or rollback to specific version via MLflow
kubectl set image deployment/wine-quality \
  wine-quality=registry/wine-quality:previous-sha