tests.yml

name: Tests
on:
  pull_request:
    paths:
      - .github/workflows/tests.yml
      - wandb_addons/**
      - pyproject.toml

jobs:
  lint:
    name: Check linting using black magic
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v3
      - uses: psf/black@stable

  run-monai-tests:
    name: Check if MonAI tests pass
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v3
      - run: |
          pip install -U pip
          pip install .[monai]
          python -c "from wandb_addons.monai import WandbStatsHandler, WandbModelCheckpointSaver"
  
  # run-keras-callbacks-tests-tf-backend:
  #   name: Check if unit-tests for the Keras-Core callbacks pass for TensorFlow backend.
  #   runs-on: ubuntu-latest
  #   env:
  #     WANDB_API_KEY: ${{ secrets.WANDB_API_KEY }}
  #   container: tensorflow/tensorflow
  #   steps:
  #     - uses: actions/checkout@v3
  #     - run: |
  #         apt-get update -y
  #         apt-get install git -y
  #         pip install -U pip
  #         pip install ".[keras, test]"
  #         KERAS_BACKEND="tensorflow" pytest -s ./wandb_addons/tests/keras/

  # run-keras-callbacks-tests-torch-backend:
  #   name: Check if unit-tests for the Keras-Core callbacks pass for PyTorch backend.
  #   runs-on: ubuntu-latest
  #   env:
  #     WANDB_API_KEY: ${{ secrets.WANDB_API_KEY }}
  #   container: pytorch/pytorch
  #   steps:
  #     - uses: actions/checkout@v3
  #     - run: |
  #         apt-get update -y
  #         apt-get install git -y
  #         pip install -U pip tensorflow-cpu
  #         pip install ".[keras, test]"
  #         KERAS_BACKEND="torch" pytest -s ./wandb_addons/tests/keras/
  
  # run-keras-callbacks-tests-jax-backend:
  #   name: Check if unit-tests for the Keras-Core callbacks pass for Jax backend.
  #   runs-on: ubuntu-latest
  #   env:
  #     WANDB_API_KEY: ${{ secrets.WANDB_API_KEY }}
  #   container: tensorflow/tensorflow
  #   steps:
  #     - uses: actions/checkout@v3
  #     - run: |
  #         apt-get update -y
  #         apt-get install git -y
  #         pip install -U pip
  #         pip install "jax[cpu]"
  #         pip install ".[keras, test]"
  #         KERAS_BACKEND="jax" pytest -s ./wandb_addons/tests/keras/

  run-jax-tests:
    name: Check if JAX, Flax and Ciclo tests pass
    runs-on: ubuntu-latest
    container: python
    steps:
      - uses: actions/checkout@v3
      - run: |
          pip install -U pip
          pip install .[jax]
          python -c "import jax"
          python -c "import flax"
          python -c "import ciclo"
          python -c "from wandb_addons.ciclo import WandbLogger"