A much faster version of nbstripout by writing it in rust (of course). This helps strip Jupyter Notebook output and metadata from notebooks. It is very useful as a git filter and is highly configurable.
pip install nbstripout-fast
Then replace nbstripout-fast with anywhere you use nbstripout.
- While we mirrored most of nbstripout's API, we do not support every nbstripout option.
- There is no CLI option to install this in git for you
- We support repository level settings in a
.git-nbconfig.yaml
file. Check out ourexamples
. On a high level, you can add a git filter in a sitewide/user level and then allow each project to enforce consistent settings.
nbstripout is a excellent project, but the python startup and import time makes its usage at scale a bit painful. While this means giving up on using nbconvert under the hood and ensuring the notebook is the correct format, it does make things up to 200x faster. This matters when you have a large number of files and git filter is called sometimes more than once per file. Let's look at the data:
Cells | nbstripout | nbstripout_fast |
---|---|---|
1 | 0m0.266s | 0m0.003s |
10 | 0m0.258s | 0m0.003s |
100 | 0m0.280s | 0m0.004s |
1000 | 0m0.372s | 0m0.013s |
10000 | 0m1.649s | 0m0.133s |
The table above shows a large overhead per notebook (mostly python startup time). When you have 100 or more notebooks, nbstripout takes more than 40s while nbstripout-fast takes only 1s!
This example illustrates how nbstripout-fast
can be used to automatically clean Jupyter notebooks using Git filters (see e.g. Git Attributes). This keeps your repository clean by removing unnecessary output and clutter, while preserving your local working version. The benefits are minimised diffs and reduced repository size.
-
Install
nbstripout-fast
as described above. -
Configure nbstripout-fast
Create a
.git-nbconfig.yaml
file at the root of your repository to configurenbstripout-fast
, e.g.nbstripout_fast: keep_count: false keep_output: false drop_empty_cells: true extra_keys: [] keep_keys: []
-
Set Git Attributes
Create a
.gitattributes
file at the root of your repository if it doesn't yet exist and add this line:*.ipynb filter=jupyter
This instructs Git to use a custom filter named "jupyter" on all
.ipynb
files. -
Configure the
jupyter
FilterRun these commands in your terminal to configure the "jupyter" filter:
git config filter.jupyter.clean nbstripout-fast git config filter.jupyter.smudge cat
clean
: This filter runsnbstripout-fast
when adding notebooks to the version that is checked out, i.e. the clean version.smudge
: This filter runscat
when checking out notebooks, ensuring your local (smudged) version remains unmodified. Git filters transform files at the time of checkout and commit.
-
Reapply Cleaning to Existing Notebooks (Optional)
If you already have Jupyter notebooks tracked by Git, you can reapply the cleaning process to them:
git add --renormalize . git commit -m "Cleaned Jupyter notebooks"
You can use cargo which will build + run the CLI:
cargo run -- -t examples/example.ipynb
You can also build with cargo and run the script with the full path:
cargo build # dev build - ./target/debug/nbstripout-fast
cargo build --release # release build - ./target/release/nbstripout-fast
Running unit tests: maturin builds this repo to include pyo3 bindings by default. This allows for us to have an extension python extension mode as well. As of today, we can't have a binary and an extension, so we use the extension only for testing (issue).
pip install -e .
maturin develop
# Should output, this way you can use RUST_LOG=debug
in-venv pytest -rP
Use RUST_LOG=debug to debug script for example:
RUST_LOG=debug cargo run -- '--extra-keys "metadata.bar cell.baz" -t foo.ipynb'
Manylinux, macos, and windows wheels and sdist are built by github workflows. Builds are triggered upon the creation of a pull request, creating a new release, or with a manual workflow dispatch. The wheels and sdist are only uploaded to PyPI when a new release is published. In order to create a new release:
- Create a commit updating the version in
Cargo.toml
andCHANGELOG.md
, then create a git tag:
git tag vX.Y.Z
git push --tags
- Draft a new release in github; select the tag that you just created.
- Once the new release is created, the wheels and sdist will be built by a
github workflow and then uploaded to PyPI automatically using the
PYPI_API_TOKEN
in the github secrets for the repository.
This plugin was contributed back to the community by the D. E. Shaw group.
This project is released under a BSD-3-Clause license.
We love contributions! Before you can contribute, please sign and submit this Contributor License Agreement (CLA). This CLA is in place to protect all users of this project.