test_pkg_on_colab.yaml

name: Test that tutorial runs on latest colab image

on:
    # push:
    #     branches: [main]
    pull_request:
        branches: [main]
    schedule:
        - cron: '0 2 3 * *'

jobs:
    test-tutorial-on-colab:
        name: Test tutorial on latest colab image
        runs-on: ubuntu-latest-4core # increase disk space  
        # https://console.cloud.google.com/artifacts/docker/colab-images/europe/public/runtime
        container:
            image: europe-docker.pkg.dev/colab-images/public/runtime:latest
        steps:
        - uses: actions/checkout@v4
        - name: Install pimms-learn (from branch) and papermill
          if: github.event_name == 'pull_request'
          run: |
            echo $GITHUB_REF
            pip install git+https://github.com/RasmussenLab/pimms.git@$GITHUB_REF
            pip install papermill
            cd project
            echo python3 -c "import pimmslearn; print(pimmslearn.__version__)"
            python3 -c "import pimmslearn; print(pimmslearn.__version__)"
        - name: Install pimms-learn (from PyPI) and papermill
          if: github.event_name == 'schedule'
          run: | 
            pip install pimms-learn papermill
        - name: Run tutorial
          run: |
            cd project
            which papermill
            papermill 04_1_train_pimms_models.ipynb 04_1_train_pimms_models_output.ipynb
            papermill 04_1_train_pimms_models.ipynb 04_1_train_pimms_models_no_val.ipynb -p sample_splits False