python.yml

name: "Build and Tests"

on:
  push:
    branches: [main]
    paths:
      - "**.py"
      - ".github/workflows/python.yml"
  pull_request:
    branches: [main]
    paths:
      - "**.py"
      - ".github/workflows/python.yml"
  release:
    types: [created]
  schedule:
    - cron: "0 0 * * 0"
  workflow_dispatch:

jobs:
  build:
    runs-on: ${{ matrix.os }}

    strategy:
      matrix:
        python-version: ["3.10", "3.11", "3.12"]
        os: [ubuntu-latest, macos-latest]

    steps:
      - uses: actions/checkout@v4

      - name: Setup Python ${{ matrix.python-version }}
        uses: actions/setup-python@v5
        with:
          python-version: ${{ matrix.python-version }}
          cache: "pip"
          cache-dependency-path: "pyproject.toml"

      - name: Install dependencies
        run: |
          pip install -U pip
          pip install jax flax fire pytest ruff
          pip install git+https://github.com/black-forest-labs/flux.git

      - name: ruff
        run: ruff check jflux
      
      - name: Test with PyTest
        run: |
          pytest -v -s .