Jupyter Notebook Output Cleaning Script

clean_nbs.py

#!/usr/bin/env ipython
import io
import sys

from nbformat import read, write

# Handle either stdin or a filename
if __name__ == '__main__':
  for filename in sys.argv[1:]:
    ipynb = read(filename, 4)

    ipynb.metadata.signature = ''

    for cell in ipynb.cells:
      if "outputs" in cell:
        cell["outputs"] = []
      if "execution_count" in cell:
        cell["execution_count"] = None

    with io.open(filename, mode="w", encoding='utf-8') as fh:
      write(ipynb, fh)