Skip to content

Feature/unload ipython extension#379

Open
nikolas-claussen wants to merge 2 commits intopatrick-kidger:mainfrom
nikolas-claussen:feature/unload-ipython-extension
Open

Feature/unload ipython extension#379
nikolas-claussen wants to merge 2 commits intopatrick-kidger:mainfrom
nikolas-claussen:feature/unload-ipython-extension

Conversation

@nikolas-claussen
Copy link

Summary

This PR adds the %unload_ext jaxtyping line magic to unload the jaxtyping extension in a Jupyter notebook. This way, users can disable type checking without having to restart their notebook kernels. This is useful for executing cells where getting type annotations is not practical.

Changes

  • jaxtyping/_ipython_extension.py:

    • Adds function unload_ipython_extension
  • jaxtyping/__init__.py:

    • Imports unload_ipython_extension
  • test/test_ipython_extension.py

    • Adds function test_unload_extension_disables_typechecking

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant