diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index c93bdcb0a877..64f8f0f4b5a7 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -9,7 +9,7 @@ repos: - repo: https://github.com/pre-commit/pre-commit-hooks - rev: v4.4.0 + rev: v4.5.0 hooks: - id: end-of-file-fixer # only include python files @@ -27,7 +27,7 @@ repos: - id: flake8 - repo: https://github.com/pre-commit/mirrors-mypy - rev: 'v1.6.0' + rev: 'v1.6.1' hooks: - id: mypy files: (jax/|tests/typing_test\.py) @@ -36,7 +36,7 @@ repos: args: [--config=pyproject.toml] - repo: https://github.com/mwouts/jupytext - rev: v1.14.7 + rev: v1.15.2 hooks: - id: jupytext args: [--sync] diff --git a/docs/autodidax.md b/docs/autodidax.md index a718c0c57156..42906c22e58d 100644 --- a/docs/autodidax.md +++ b/docs/autodidax.md @@ -6,7 +6,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/autodidax.py b/docs/autodidax.py index 199cce92b40d..784b28b3a177 100644 --- a/docs/autodidax.py +++ b/docs/autodidax.py @@ -20,7 +20,7 @@ # extension: .py # format_name: light # format_version: '1.5' -# jupytext_version: 1.14.7 +# jupytext_version: 1.15.2 # kernelspec: # display_name: Python 3 # name: python3 diff --git a/docs/developer.md b/docs/developer.md index 1896b1afc78f..c4df66c83f41 100644 --- a/docs/developer.md +++ b/docs/developer.md @@ -410,7 +410,7 @@ using [jupytext](https://jupytext.readthedocs.io/) by running `jupytext --sync` notebooks; for example: ``` -pip install jupytext==1.14.7 +pip install jupytext==1.15.2 jupytext --sync docs/notebooks/quickstart.ipynb ``` diff --git a/docs/jax-101/01-jax-basics.md b/docs/jax-101/01-jax-basics.md index f2ac47cdc1cb..b283fdc8a516 100644 --- a/docs/jax-101/01-jax-basics.md +++ b/docs/jax-101/01-jax-basics.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/jax-101/02-jitting.md b/docs/jax-101/02-jitting.md index b3eab7c1035f..023d94c3685d 100644 --- a/docs/jax-101/02-jitting.md +++ b/docs/jax-101/02-jitting.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/jax-101/03-vectorization.md b/docs/jax-101/03-vectorization.md index 0df3c8706cc2..4e6049a36752 100644 --- a/docs/jax-101/03-vectorization.md +++ b/docs/jax-101/03-vectorization.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/jax-101/04-advanced-autodiff.md b/docs/jax-101/04-advanced-autodiff.md index d7f3250cd0d7..e0b36dfca531 100644 --- a/docs/jax-101/04-advanced-autodiff.md +++ b/docs/jax-101/04-advanced-autodiff.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/jax-101/05-random-numbers.md b/docs/jax-101/05-random-numbers.md index 90e97f697ff8..e52b2f3268df 100644 --- a/docs/jax-101/05-random-numbers.md +++ b/docs/jax-101/05-random-numbers.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/jax-101/05.1-pytrees.md b/docs/jax-101/05.1-pytrees.md index b806eacd8a4d..3ac3a8c73b08 100644 --- a/docs/jax-101/05.1-pytrees.md +++ b/docs/jax-101/05.1-pytrees.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/jax-101/06-parallelism.md b/docs/jax-101/06-parallelism.md index a2923610bc98..e7d08cfc74e8 100644 --- a/docs/jax-101/06-parallelism.md +++ b/docs/jax-101/06-parallelism.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/jax-101/07-state.md b/docs/jax-101/07-state.md index c6afbd2151e5..5482b278f4c2 100644 --- a/docs/jax-101/07-state.md +++ b/docs/jax-101/07-state.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/jep/9407-type-promotion.md b/docs/jep/9407-type-promotion.md index b6a5d4fc2256..06cb325afed8 100644 --- a/docs/jep/9407-type-promotion.md +++ b/docs/jep/9407-type-promotion.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 language: python diff --git a/docs/notebooks/Common_Gotchas_in_JAX.md b/docs/notebooks/Common_Gotchas_in_JAX.md index 619ebcc630b9..f2fcd669cd6c 100644 --- a/docs/notebooks/Common_Gotchas_in_JAX.md +++ b/docs/notebooks/Common_Gotchas_in_JAX.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 language: python diff --git a/docs/notebooks/Custom_derivative_rules_for_Python_code.md b/docs/notebooks/Custom_derivative_rules_for_Python_code.md index 003ea18edc54..6af44f92d468 100644 --- a/docs/notebooks/Custom_derivative_rules_for_Python_code.md +++ b/docs/notebooks/Custom_derivative_rules_for_Python_code.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/notebooks/Distributed_arrays_and_automatic_parallelization.md b/docs/notebooks/Distributed_arrays_and_automatic_parallelization.md index 3d0392a617de..ec50b58c3e54 100644 --- a/docs/notebooks/Distributed_arrays_and_automatic_parallelization.md +++ b/docs/notebooks/Distributed_arrays_and_automatic_parallelization.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/notebooks/How_JAX_primitives_work.md b/docs/notebooks/How_JAX_primitives_work.md index 0ee55f00c97f..b8da00edb240 100644 --- a/docs/notebooks/How_JAX_primitives_work.md +++ b/docs/notebooks/How_JAX_primitives_work.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/notebooks/Neural_Network_and_Data_Loading.md b/docs/notebooks/Neural_Network_and_Data_Loading.md index df99260de16e..37d2521763df 100644 --- a/docs/notebooks/Neural_Network_and_Data_Loading.md +++ b/docs/notebooks/Neural_Network_and_Data_Loading.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 language: python diff --git a/docs/notebooks/Writing_custom_interpreters_in_Jax.md b/docs/notebooks/Writing_custom_interpreters_in_Jax.md index 684393d6d975..7de91d4844b9 100644 --- a/docs/notebooks/Writing_custom_interpreters_in_Jax.md +++ b/docs/notebooks/Writing_custom_interpreters_in_Jax.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 language: python diff --git a/docs/notebooks/autodiff_cookbook.md b/docs/notebooks/autodiff_cookbook.md index 6df039196b5f..dfb48cdebc18 100644 --- a/docs/notebooks/autodiff_cookbook.md +++ b/docs/notebooks/autodiff_cookbook.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 language: python diff --git a/docs/notebooks/autodiff_remat.md b/docs/notebooks/autodiff_remat.md index d8486a0b532f..37c444afa581 100644 --- a/docs/notebooks/autodiff_remat.md +++ b/docs/notebooks/autodiff_remat.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/notebooks/convolutions.md b/docs/notebooks/convolutions.md index ce5b5a4aa576..b29e0e6dcd92 100644 --- a/docs/notebooks/convolutions.md +++ b/docs/notebooks/convolutions.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 language: python diff --git a/docs/notebooks/external_callbacks.md b/docs/notebooks/external_callbacks.md index e87c360f5706..4dda2f9b404a 100644 --- a/docs/notebooks/external_callbacks.md +++ b/docs/notebooks/external_callbacks.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/notebooks/neural_network_with_tfds_data.md b/docs/notebooks/neural_network_with_tfds_data.md index 7bf1a92fff12..b0f1275843c5 100644 --- a/docs/notebooks/neural_network_with_tfds_data.md +++ b/docs/notebooks/neural_network_with_tfds_data.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 language: python diff --git a/docs/notebooks/quickstart.md b/docs/notebooks/quickstart.md index fa2f5cbd9d4d..24f4990aceed 100644 --- a/docs/notebooks/quickstart.md +++ b/docs/notebooks/quickstart.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 language: python diff --git a/docs/notebooks/thinking_in_jax.md b/docs/notebooks/thinking_in_jax.md index 6a416f83ad80..7d0818a78b20 100644 --- a/docs/notebooks/thinking_in_jax.md +++ b/docs/notebooks/thinking_in_jax.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/notebooks/vmapped_log_probs.md b/docs/notebooks/vmapped_log_probs.md index 3564f3db9537..74a07264c0f6 100644 --- a/docs/notebooks/vmapped_log_probs.md +++ b/docs/notebooks/vmapped_log_probs.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 language: python diff --git a/docs/notebooks/xmap_tutorial.md b/docs/notebooks/xmap_tutorial.md index cb13fa6b4afe..4dda13b44d42 100644 --- a/docs/notebooks/xmap_tutorial.md +++ b/docs/notebooks/xmap_tutorial.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 name: python3 diff --git a/docs/pallas/quickstart.md b/docs/pallas/quickstart.md index 25b1a5899990..1e7e61721c07 100644 --- a/docs/pallas/quickstart.md +++ b/docs/pallas/quickstart.md @@ -5,7 +5,7 @@ jupytext: extension: .md format_name: myst format_version: 0.13 - jupytext_version: 1.14.7 + jupytext_version: 1.15.2 kernelspec: display_name: Python 3 (ipykernel) language: python