Simple Python script to fail CI/CD checks if Jupyter notebooks aren't cleared

check_notebooks_cleared.py

import glob
import subprocess
import sys

CLEAR_COMMAND = "jupyter nbconvert --to notebook --inplace --clear-output"
DIFF_COMMAND = "git diff --quiet --exit-code"

# Clear notebook outputs
for notebook_file in glob.glob("notebooks/*.ipynb"):
    subprocess.run([*CLEAR_COMMAND.split(), notebook_file])

# Check if clearing notebooks changed anything
exit_code = subprocess.run([*DIFF_COMMAND.split()]).returncode

# Fail if it did
sys.exit(exit_code)