unittest.yml

name: Python Unit Tests

# Defines when the action should run. This example triggers on push and pull requests to the main branch.
on:
  push:
    branches: [ main ]
  pull_request:
    branches: [ main ]

# Jobs that the workflow will execute
jobs:
  run-unittests:
    # The type of runner that the job will run on
    runs-on: ubuntu-latest
    strategy:
      matrix:
        python-version: ["3.10"]

    # Steps represent a sequence of tasks that will be executed as part of the job
    steps:
    # Checks-out your repository under $GITHUB_WORKSPACE, so your job can access it
    - uses: actions/checkout@v3

    # Sets up a Python environment
    - name: Set up Python ${{ matrix.python-version }}
      uses: actions/setup-python@v4
      with:
        python-version: ${{ matrix.python-version }}

    # Install dependencies (if any)
    - name: Install dependencies
      run: |
        python -m pip install --upgrade pip
        pip install torch==1.13.1 torchvision==0.14.1 jax==0.4.23 jaxlib==0.4.23 diffusers==0.18.2 joblib

    # Run unittests using the Python unittest module
    - name: Run unittest
      run: python -m unittest discover