pymc-marketing/.github/workflows/test_notebook.yml at 2db827c9164e7fe5d1dae12aae496af2790bc5b8 · pymc-labs/pymc-marketing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
name: Test Notebooks

on:
  pull_request:
    branches: [main]
    paths:
     - "pyproject.toml"
     - "tests/**.py"
     - "pymc_marketing/**"
     - ".github/workflows/test_notebook.yml"
     - "scripts/run_notebooks/**.py"
     - "docs/source/notebooks/**.ipynb"
     - "!docs/source/notebooks/mmm/**.ipynb"
     - "!docs/source/notebooks/*/dev/**.ipynb"
  push:
    branches: [main]
    paths:
     - "pyproject.toml"
     - "tests/**.py"
     - "pymc_marketing/**"
     - ".github/workflows/test_notebook.yml"
     - "scripts/run_notebooks/**.py"
     - "docs/source/notebooks/**.ipynb"
     - "!docs/source/notebooks/mmm/**.ipynb"
     - "!docs/source/notebooks/*/dev/**.ipynb"

concurrency:
  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
  cancel-in-progress: true

jobs:
  notebooks:
    runs-on: ubuntu-latest
    timeout-minutes: 60
    strategy:
      fail-fast: false
      matrix:
        linker: [cvm, numba]
        split:
          - "--notebooks mmm --end-idx 10"
          - "--notebooks mmm --start-idx 10"
          - "--notebooks clv"
          - "--notebooks bass"
          - "--notebooks customer_choice"
          - "--notebooks general"
          - "--exclude-dirs mmm clv bass customer_choice general"
    env:
      PYTENSOR_FLAGS: linker=${{ matrix.linker }}
    steps:
    - uses: actions/checkout@v6

    - name: Install uv
      uses: astral-sh/setup-uv@v7

    - name: Install dependencies
      run: |
        sudo apt-get install graphviz graphviz-dev
        uv venv --python 3.12
        uv pip install --upgrade pip
        uv pip install -e ".[docs,test,dag]"
        uv pip list

    - name: Run notebooks
      run: uv run scripts/run_notebooks/runner.py ${{ matrix.split }}