Python 3.10.12
Requirement already satisfied: pip in /tmp/chex-env/lib/python3.10/site-packages (22.0.2)
Collecting pip
  Using cached pip-23.3.2-py3-none-any.whl (2.1 MB)
Requirement already satisfied: setuptools in /tmp/chex-env/lib/python3.10/site-packages (59.6.0)
Collecting setuptools
  Using cached setuptools-69.0.3-py3-none-any.whl (819 kB)
Collecting wheel
  Using cached wheel-0.42.0-py3-none-any.whl (65 kB)
Installing collected packages: wheel, setuptools, pip
  Attempting uninstall: setuptools
    Found existing installation: setuptools 59.6.0
    Uninstalling setuptools-59.6.0:
      Successfully uninstalled setuptools-59.6.0
  Attempting uninstall: pip
    Found existing installation: pip 22.0.2
    Uninstalling pip-22.0.2:
      Successfully uninstalled pip-22.0.2
Successfully installed pip-23.3.2 setuptools-69.0.3 wheel-0.42.0
Collecting flake8
  Using cached flake8-6.1.0-py2.py3-none-any.whl.metadata (3.8 kB)
Collecting pytest-xdist
  Using cached pytest_xdist-3.5.0-py3-none-any.whl.metadata (3.1 kB)
Collecting pylint
  Using cached pylint-3.0.3-py3-none-any.whl.metadata (12 kB)
Collecting pylint-exit
  Using cached pylint_exit-1.2.0-py2.py3-none-any.whl (6.3 kB)
Collecting mccabe<0.8.0,>=0.7.0 (from flake8)
  Using cached mccabe-0.7.0-py2.py3-none-any.whl (7.3 kB)
Collecting pycodestyle<2.12.0,>=2.11.0 (from flake8)
  Using cached pycodestyle-2.11.1-py2.py3-none-any.whl.metadata (4.5 kB)
Collecting pyflakes<3.2.0,>=3.1.0 (from flake8)
  Using cached pyflakes-3.1.0-py2.py3-none-any.whl.metadata (3.5 kB)
Collecting execnet>=1.1 (from pytest-xdist)
  Using cached execnet-2.0.2-py3-none-any.whl.metadata (2.9 kB)
Collecting pytest>=6.2.0 (from pytest-xdist)
  Using cached pytest-7.4.3-py3-none-any.whl.metadata (7.9 kB)
Collecting platformdirs>=2.2.0 (from pylint)
  Using cached platformdirs-4.1.0-py3-none-any.whl.metadata (11 kB)
Collecting astroid<=3.1.0-dev0,>=3.0.1 (from pylint)
  Using cached astroid-3.0.2-py3-none-any.whl.metadata (4.5 kB)
Collecting isort!=5.13.0,<6,>=4.2.5 (from pylint)
  Using cached isort-5.13.2-py3-none-any.whl.metadata (12 kB)
Collecting tomlkit>=0.10.1 (from pylint)
  Using cached tomlkit-0.12.3-py3-none-any.whl.metadata (2.7 kB)
Collecting dill>=0.2 (from pylint)
  Using cached dill-0.3.7-py3-none-any.whl.metadata (9.9 kB)
Collecting tomli>=1.1.0 (from pylint)
  Using cached tomli-2.0.1-py3-none-any.whl (12 kB)
Collecting typing-extensions>=4.0.0 (from astroid<=3.1.0-dev0,>=3.0.1->pylint)
  Using cached typing_extensions-4.9.0-py3-none-any.whl.metadata (3.0 kB)
Collecting iniconfig (from pytest>=6.2.0->pytest-xdist)
  Using cached iniconfig-2.0.0-py3-none-any.whl (5.9 kB)
Collecting packaging (from pytest>=6.2.0->pytest-xdist)
  Using cached packaging-23.2-py3-none-any.whl.metadata (3.2 kB)
Collecting pluggy<2.0,>=0.12 (from pytest>=6.2.0->pytest-xdist)
  Using cached pluggy-1.3.0-py3-none-any.whl.metadata (4.3 kB)
Collecting exceptiongroup>=1.0.0rc8 (from pytest>=6.2.0->pytest-xdist)
  Using cached exceptiongroup-1.2.0-py3-none-any.whl.metadata (6.6 kB)
Using cached flake8-6.1.0-py2.py3-none-any.whl (58 kB)
Using cached pytest_xdist-3.5.0-py3-none-any.whl (42 kB)
Using cached pylint-3.0.3-py3-none-any.whl (510 kB)
Using cached astroid-3.0.2-py3-none-any.whl (275 kB)
Using cached dill-0.3.7-py3-none-any.whl (115 kB)
Using cached execnet-2.0.2-py3-none-any.whl (37 kB)
Using cached isort-5.13.2-py3-none-any.whl (92 kB)
Using cached platformdirs-4.1.0-py3-none-any.whl (17 kB)
Using cached pycodestyle-2.11.1-py2.py3-none-any.whl (31 kB)
Using cached pyflakes-3.1.0-py2.py3-none-any.whl (62 kB)
Using cached pytest-7.4.3-py3-none-any.whl (325 kB)
Using cached tomlkit-0.12.3-py3-none-any.whl (37 kB)
Using cached exceptiongroup-1.2.0-py3-none-any.whl (16 kB)
Using cached pluggy-1.3.0-py3-none-any.whl (18 kB)
Using cached typing_extensions-4.9.0-py3-none-any.whl (32 kB)
Using cached packaging-23.2-py3-none-any.whl (53 kB)
Installing collected packages: pylint-exit, typing-extensions, tomlkit, tomli, pyflakes, pycodestyle, pluggy, platformdirs, packaging, mccabe, isort, iniconfig, execnet, exceptiongroup, dill, pytest, flake8, astroid, pytest-xdist, pylint
Successfully installed astroid-3.0.2 dill-0.3.7 exceptiongroup-1.2.0 execnet-2.0.2 flake8-6.1.0 iniconfig-2.0.0 isort-5.13.2 mccabe-0.7.0 packaging-23.2 platformdirs-4.1.0 pluggy-1.3.0 pycodestyle-2.11.1 pyflakes-3.1.0 pylint-3.0.3 pylint-exit-1.2.0 pytest-7.4.3 pytest-xdist-3.5.0 tomli-2.0.1 tomlkit-0.12.3 typing-extensions-4.9.0
Ignoring setuptools: markers 'python_version >= "3.12"' don't match your environment
Collecting absl-py>=0.9.0 (from -r requirements/requirements.txt (line 1))
  Using cached absl_py-2.0.0-py3-none-any.whl.metadata (2.3 kB)
Requirement already satisfied: typing_extensions>=4.2.0 in /tmp/chex-env/lib/python3.10/site-packages (from -r requirements/requirements.txt (line 2)) (4.9.0)
Collecting jax>=0.4.16 (from -r requirements/requirements.txt (line 3))
  Using cached jax-0.4.23-py3-none-any.whl.metadata (24 kB)
Collecting jaxlib>=0.1.37 (from -r requirements/requirements.txt (line 4))
  Using cached jaxlib-0.4.23-cp310-cp310-manylinux2014_x86_64.whl.metadata (2.1 kB)
Collecting numpy>=1.24.1 (from -r requirements/requirements.txt (line 5))
  Using cached numpy-1.26.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (61 kB)
Collecting toolz>=0.9.0 (from -r requirements/requirements.txt (line 7))
  Using cached toolz-0.12.0-py3-none-any.whl (55 kB)
Collecting ml-dtypes>=0.2.0 (from jax>=0.4.16->-r requirements/requirements.txt (line 3))
  Using cached ml_dtypes-0.3.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (20 kB)
Collecting opt-einsum (from jax>=0.4.16->-r requirements/requirements.txt (line 3))
  Using cached opt_einsum-3.3.0-py3-none-any.whl (65 kB)
Collecting scipy>=1.9 (from jax>=0.4.16->-r requirements/requirements.txt (line 3))
  Using cached scipy-1.11.4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (60 kB)
Using cached absl_py-2.0.0-py3-none-any.whl (130 kB)
Using cached jax-0.4.23-py3-none-any.whl (1.7 MB)
Using cached jaxlib-0.4.23-cp310-cp310-manylinux2014_x86_64.whl (77.2 MB)
Using cached numpy-1.26.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (18.2 MB)
Using cached ml_dtypes-0.3.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (206 kB)
Using cached scipy-1.11.4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (36.4 MB)
Installing collected packages: toolz, numpy, absl-py, scipy, opt-einsum, ml-dtypes, jaxlib, jax
Successfully installed absl-py-2.0.0 jax-0.4.23 jaxlib-0.4.23 ml-dtypes-0.3.1 numpy-1.26.2 opt-einsum-3.3.0 scipy-1.11.4 toolz-0.12.0
Collecting cloudpickle==2.2.0 (from -r requirements/requirements-test.txt (line 1))
  Using cached cloudpickle-2.2.0-py3-none-any.whl (25 kB)
Collecting dm-tree>=0.1.5 (from -r requirements/requirements-test.txt (line 2))
  Using cached dm_tree-0.1.8-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (152 kB)
Installing collected packages: dm-tree, cloudpickle
Successfully installed cloudpickle-2.2.0 dm-tree-0.1.8
0

------------------------------------
Your code has been rated at 10.00/10


------------------------------------
Your code has been rated at 10.00/10

running sdist
running egg_info
creating chex.egg-info
writing chex.egg-info/PKG-INFO
writing dependency_links to chex.egg-info/dependency_links.txt
writing requirements to chex.egg-info/requires.txt
writing top-level names to chex.egg-info/top_level.txt
writing manifest file 'chex.egg-info/SOURCES.txt'
reading manifest file 'chex.egg-info/SOURCES.txt'
reading manifest template 'MANIFEST.in'
adding license file 'LICENSE'
writing manifest file 'chex.egg-info/SOURCES.txt'
running check
creating chex-0.1.85
creating chex-0.1.85/chex
creating chex-0.1.85/chex.egg-info
creating chex-0.1.85/chex/_src
creating chex-0.1.85/requirements
copying files to chex-0.1.85...
copying LICENSE -> chex-0.1.85
copying MANIFEST.in -> chex-0.1.85
copying README.md -> chex-0.1.85
copying setup.py -> chex-0.1.85
copying chex/__init__.py -> chex-0.1.85/chex
copying chex/chex_test.py -> chex-0.1.85/chex
copying chex/py.typed -> chex-0.1.85/chex
copying chex.egg-info/PKG-INFO -> chex-0.1.85/chex.egg-info
copying chex.egg-info/SOURCES.txt -> chex-0.1.85/chex.egg-info
copying chex.egg-info/dependency_links.txt -> chex-0.1.85/chex.egg-info
copying chex.egg-info/not-zip-safe -> chex-0.1.85/chex.egg-info
copying chex.egg-info/requires.txt -> chex-0.1.85/chex.egg-info
copying chex.egg-info/top_level.txt -> chex-0.1.85/chex.egg-info
copying chex/_src/__init__.py -> chex-0.1.85/chex/_src
copying chex/_src/asserts.py -> chex-0.1.85/chex/_src
copying chex/_src/asserts_chexify.py -> chex-0.1.85/chex/_src
copying chex/_src/asserts_chexify_test.py -> chex-0.1.85/chex/_src
copying chex/_src/asserts_internal.py -> chex-0.1.85/chex/_src
copying chex/_src/asserts_internal_test.py -> chex-0.1.85/chex/_src
copying chex/_src/asserts_test.py -> chex-0.1.85/chex/_src
copying chex/_src/dataclass.py -> chex-0.1.85/chex/_src
copying chex/_src/dataclass_test.py -> chex-0.1.85/chex/_src
copying chex/_src/dimensions.py -> chex-0.1.85/chex/_src
copying chex/_src/dimensions_test.py -> chex-0.1.85/chex/_src
copying chex/_src/fake.py -> chex-0.1.85/chex/_src
copying chex/_src/fake_set_n_cpu_devices_test.py -> chex-0.1.85/chex/_src
copying chex/_src/fake_test.py -> chex-0.1.85/chex/_src
copying chex/_src/pytypes.py -> chex-0.1.85/chex/_src
copying chex/_src/restrict_backends.py -> chex-0.1.85/chex/_src
copying chex/_src/restrict_backends_test.py -> chex-0.1.85/chex/_src
copying chex/_src/variants.py -> chex-0.1.85/chex/_src
copying chex/_src/variants_test.py -> chex-0.1.85/chex/_src
copying requirements/requirements-docs.txt -> chex-0.1.85/requirements
copying requirements/requirements-test.txt -> chex-0.1.85/requirements
copying requirements/requirements.txt -> chex-0.1.85/requirements
copying chex.egg-info/SOURCES.txt -> chex-0.1.85/chex.egg-info
Writing chex-0.1.85/setup.cfg
creating dist
Creating tar archive
removing 'chex-0.1.85' (and everything under it)
Processing ./dist/chex-0.1.85.tar.gz
  Preparing metadata (setup.py): started
  Preparing metadata (setup.py): finished with status 'done'
Building wheels for collected packages: chex
  Building wheel for chex (setup.py): started
  Building wheel for chex (setup.py): finished with status 'done'
  Created wheel for chex: filename=chex-0.1.85-py3-none-any.whl size=95175 sha256=99fd12336faceed9336a12b6ba7f1aa35a59e8b10763419b08b8cbe08aec6e2f
  Stored in directory: /home/mishmish/.cache/pip/wheels/42/09/a1/08872f41318204d35628f82e5b6abbe1cbe15e02f93be8b426
Successfully built chex
Processing ./chex-0.1.85-py3-none-any.whl
Requirement already satisfied: absl-py>=0.9.0 in /tmp/chex-env/lib/python3.10/site-packages (from chex==0.1.85) (2.0.0)
Requirement already satisfied: typing-extensions>=4.2.0 in /tmp/chex-env/lib/python3.10/site-packages (from chex==0.1.85) (4.9.0)
Requirement already satisfied: jax>=0.4.16 in /tmp/chex-env/lib/python3.10/site-packages (from chex==0.1.85) (0.4.23)
Requirement already satisfied: jaxlib>=0.1.37 in /tmp/chex-env/lib/python3.10/site-packages (from chex==0.1.85) (0.4.23)
Requirement already satisfied: numpy>=1.24.1 in /tmp/chex-env/lib/python3.10/site-packages (from chex==0.1.85) (1.26.2)
Requirement already satisfied: toolz>=0.9.0 in /tmp/chex-env/lib/python3.10/site-packages (from chex==0.1.85) (0.12.0)
Requirement already satisfied: ml-dtypes>=0.2.0 in /tmp/chex-env/lib/python3.10/site-packages (from jax>=0.4.16->chex==0.1.85) (0.3.1)
Requirement already satisfied: opt-einsum in /tmp/chex-env/lib/python3.10/site-packages (from jax>=0.4.16->chex==0.1.85) (3.3.0)
Requirement already satisfied: scipy>=1.9 in /tmp/chex-env/lib/python3.10/site-packages (from jax>=0.4.16->chex==0.1.85) (1.11.4)
Installing collected packages: chex
Successfully installed chex-0.1.85
Collecting pytype
  Using cached pytype-2023.12.18-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (10 kB)
Collecting attrs>=21.4.0 (from pytype)
  Using cached attrs-23.1.0-py3-none-any.whl (61 kB)
Collecting importlab>=0.8 (from pytype)
  Using cached importlab-0.8.1-py2.py3-none-any.whl.metadata (2.4 kB)
Collecting jinja2>=3.1.2 (from pytype)
  Using cached Jinja2-3.1.2-py3-none-any.whl (133 kB)
Collecting libcst>=1.0.1 (from pytype)
  Using cached libcst-1.1.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (17 kB)
Collecting networkx<3.2 (from pytype)
  Using cached networkx-3.1-py3-none-any.whl (2.1 MB)
Collecting ninja>=1.10.0.post2 (from pytype)
  Using cached ninja-1.11.1.1-py2.py3-none-manylinux1_x86_64.manylinux_2_5_x86_64.whl.metadata (5.3 kB)
Collecting pycnite>=2023.10.11 (from pytype)
  Using cached pycnite-2023.10.11-py3-none-any.whl.metadata (1.0 kB)
Collecting pydot>=1.4.2 (from pytype)
  Using cached pydot-1.4.2-py2.py3-none-any.whl (21 kB)
Collecting tabulate>=0.8.10 (from pytype)
  Using cached tabulate-0.9.0-py3-none-any.whl (35 kB)
Collecting toml>=0.10.2 (from pytype)
  Using cached toml-0.10.2-py2.py3-none-any.whl (16 kB)
Requirement already satisfied: typing-extensions>=4.3.0 in /tmp/chex-env/lib/python3.10/site-packages (from pytype) (4.9.0)
Collecting MarkupSafe>=2.0 (from jinja2>=3.1.2->pytype)
  Using cached MarkupSafe-2.1.3-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (3.0 kB)
Collecting typing-inspect>=0.4.0 (from libcst>=1.0.1->pytype)
  Using cached typing_inspect-0.9.0-py3-none-any.whl.metadata (1.5 kB)
Collecting pyyaml>=5.2 (from libcst>=1.0.1->pytype)
  Using cached PyYAML-6.0.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (2.1 kB)
Collecting pyparsing>=2.1.4 (from pydot>=1.4.2->pytype)
  Using cached pyparsing-3.1.1-py3-none-any.whl.metadata (5.1 kB)
Collecting mypy-extensions>=0.3.0 (from typing-inspect>=0.4.0->libcst>=1.0.1->pytype)
  Using cached mypy_extensions-1.0.0-py3-none-any.whl (4.7 kB)
Using cached pytype-2023.12.18-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.3 MB)
Using cached importlab-0.8.1-py2.py3-none-any.whl (21 kB)
Using cached libcst-1.1.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (3.2 MB)
Using cached ninja-1.11.1.1-py2.py3-none-manylinux1_x86_64.manylinux_2_5_x86_64.whl (307 kB)
Using cached pycnite-2023.10.11-py3-none-any.whl (21 kB)
Using cached MarkupSafe-2.1.3-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (25 kB)
Using cached pyparsing-3.1.1-py3-none-any.whl (103 kB)
Using cached PyYAML-6.0.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (705 kB)
Using cached typing_inspect-0.9.0-py3-none-any.whl (8.8 kB)
Installing collected packages: ninja, toml, tabulate, pyyaml, pyparsing, pycnite, networkx, mypy-extensions, MarkupSafe, attrs, typing-inspect, pydot, jinja2, importlab, libcst, pytype
Successfully installed MarkupSafe-2.1.3 attrs-23.1.0 importlab-0.8.1 jinja2-3.1.2 libcst-1.1.0 mypy-extensions-1.0.0 networkx-3.1 ninja-1.11.1.1 pycnite-2023.10.11 pydot-1.4.2 pyparsing-3.1.1 pytype-2023.12.18 pyyaml-6.0.1 tabulate-0.9.0 toml-0.10.2 typing-inspect-0.9.0
ninja: Entering directory `.pytype'
[1/19] check chex._src.pytypes
[2/19] check chex._src.fake
[3/19] check chex._src.asserts_internal
[4/19] check chex._src.variants
[5/19] check chex._src.asserts
[6/19] check chex._src.asserts_chexify
[7/19] check chex._src.restrict_backends
[8/19] check chex._src.dataclass
[9/19] check chex._src.dimensions
[10/19] check chex._src.__init__
[11/19] check chex._src.variants_test
[12/19] check chex._src.asserts_chexify_test
[13/19] check chex._src.restrict_backends_test
[14/19] check chex._src.dataclass_test
[15/19] check chex._src.asserts_internal_test
[16/19] check chex._src.dimensions_test
[17/19] check chex._src.fake_test
[18/19] check chex._src.fake_set_n_cpu_devices_test
[19/19] check chex._src.asserts_test
Computing dependencies
Analyzing 19 sources with 0 local dependencies
Leaving directory '.pytype'
Success: no errors found
Requirement already satisfied: cloudpickle==2.2.0 in /tmp/chex-env/lib/python3.10/site-packages (from -r requirements/requirements-test.txt (line 1)) (2.2.0)
Requirement already satisfied: dm-tree>=0.1.5 in /tmp/chex-env/lib/python3.10/site-packages (from -r requirements/requirements-test.txt (line 2)) (0.1.8)
============================= test session starts ==============================
platform linux -- Python 3.10.12, pytest-7.4.3, pluggy-1.3.0
rootdir: /home/mishmish/Documents/chex
plugins: xdist-3.5.0
created: 16/16 workers
16 workers [607 items]

.sss.........ss..ss..ss.s.s...ss.s.s...s...s......s.s................... [ 11%]
......................................................................... [ 23%]
.................................s...ss.s..ss.s...sss..ss..s.ss..ss..s.ss [ 35%]
...s..ss..........s...s...ss....s......................................... [ 47%]
.......s................................................................ [ 59%]
..........................................................sss.....ss..ss [ 71%]
..s............s..ss....s..............s..s............................. [ 83%]
...............................s..ssssssssss.sssssss.sss................ [ 95%]
...........s...............                                              [100%]
=============================== warnings summary ===============================
_src/asserts_test.py: 13 warnings
  /tmp/chex-env/lib/python3.10/site-packages/chex/_src/asserts.py:1182: DeprecationWarning: arr.device() is deprecated. Use arr.devices() instead.
    if leaf.device().platform not in platform:

_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_device
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_device
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_device
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_device
  /tmp/chex-env/lib/python3.10/site-packages/chex/_src/asserts.py:1188: DeprecationWarning: arr.device() is deprecated. Use arr.devices() instead.
    if device is not None and leaf.device() != device:

_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_device
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_device
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_device
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_device
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_device
  /tmp/chex-env/lib/python3.10/site-packages/chex/_src/asserts.py:1185: DeprecationWarning: arr.device() is deprecated. Use arr.devices() instead.
    f"'{leaf.device().platform}', expected '{platform}'."))

_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_host
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_host
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_host
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_host
  /tmp/chex-env/lib/python3.10/site-packages/chex/_src/asserts.py:1127: DeprecationWarning: arr.device() is deprecated. Use arr.devices() instead.
    f"on {leaf.device()} (CPU devices are disallowed)."))

_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_host
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_on_host
  /tmp/chex-env/lib/python3.10/site-packages/chex/_src/asserts.py:1120: DeprecationWarning: arr.device() is deprecated. Use arr.devices() instead.
    if leaf.device().platform != "cpu":

_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_sharded
_src/asserts_test.py::TreeAssertionsTest::test_assert_tree_is_sharded
  /tmp/chex-env/lib/python3.10/site-packages/chex/_src/asserts.py:1238: DeprecationWarning: arr.device() is deprecated. Use arr.devices() instead.
    f" (device={leaf.device()})."

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
================ 523 passed, 84 skipped, 30 warnings in 15.82s =================
============================= test session starts ==============================
platform linux -- Python 3.10.12, pytest-7.4.3, pluggy-1.3.0
rootdir: /home/mishmish/Documents/chex
plugins: xdist-3.5.0
created: 16/16 workers
16 workers [1 item]

.                                                                        [100%]
============================== 1 passed in 1.34s ===============================
Collecting sphinx==4.5.0 (from -r requirements/requirements-docs.txt (line 1))
  Using cached Sphinx-4.5.0-py3-none-any.whl (3.1 MB)
Collecting sphinx_rtd_theme==1.0.0 (from -r requirements/requirements-docs.txt (line 2))
  Using cached sphinx_rtd_theme-1.0.0-py2.py3-none-any.whl (2.8 MB)
Collecting sphinxcontrib-katex==0.8.6 (from -r requirements/requirements-docs.txt (line 3))
  Using cached sphinxcontrib_katex-0.8.6-py3-none-any.whl (10 kB)
Collecting sphinx-autodoc-typehints==1.11.1 (from -r requirements/requirements-docs.txt (line 4))
  Using cached sphinx_autodoc_typehints-1.11.1-py3-none-any.whl (8.7 kB)
Collecting IPython==7.16.3 (from -r requirements/requirements-docs.txt (line 5))
  Using cached ipython-7.16.3-py3-none-any.whl (783 kB)
Collecting ipykernel==5.3.4 (from -r requirements/requirements-docs.txt (line 6))
  Using cached ipykernel-5.3.4-py3-none-any.whl (120 kB)
Collecting pandoc==1.0.2 (from -r requirements/requirements-docs.txt (line 7))
  Using cached pandoc-1.0.2-py3-none-any.whl
Collecting myst_nb==0.13.1 (from -r requirements/requirements-docs.txt (line 8))
  Using cached myst_nb-0.13.1-py3-none-any.whl (37 kB)
Collecting docutils==0.16 (from -r requirements/requirements-docs.txt (line 9))
  Using cached docutils-0.16-py2.py3-none-any.whl (548 kB)
Collecting matplotlib==3.5.0 (from -r requirements/requirements-docs.txt (line 10))
  Using cached matplotlib-3.5.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (11.9 MB)
Collecting sphinxcontrib-applehelp (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached sphinxcontrib_applehelp-1.0.7-py3-none-any.whl.metadata (2.2 kB)
Collecting sphinxcontrib-devhelp (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached sphinxcontrib_devhelp-1.0.5-py3-none-any.whl.metadata (2.2 kB)
Collecting sphinxcontrib-jsmath (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached sphinxcontrib_jsmath-1.0.1-py2.py3-none-any.whl (5.1 kB)
Collecting sphinxcontrib-htmlhelp>=2.0.0 (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached sphinxcontrib_htmlhelp-2.0.4-py3-none-any.whl.metadata (2.2 kB)
Collecting sphinxcontrib-serializinghtml>=1.1.5 (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached sphinxcontrib_serializinghtml-1.1.9-py3-none-any.whl.metadata (2.3 kB)
Collecting sphinxcontrib-qthelp (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached sphinxcontrib_qthelp-1.0.6-py3-none-any.whl.metadata (2.2 kB)
Requirement already satisfied: Jinja2>=2.3 in /tmp/chex-env/lib/python3.10/site-packages (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1)) (3.1.2)
Collecting Pygments>=2.0 (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached pygments-2.17.2-py3-none-any.whl.metadata (2.6 kB)
Collecting snowballstemmer>=1.1 (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached snowballstemmer-2.2.0-py2.py3-none-any.whl (93 kB)
Collecting babel>=1.3 (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached Babel-2.14.0-py3-none-any.whl.metadata (1.6 kB)
Collecting alabaster<0.8,>=0.7 (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached alabaster-0.7.13-py3-none-any.whl (13 kB)
Collecting imagesize (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached imagesize-1.4.1-py2.py3-none-any.whl (8.8 kB)
Collecting requests>=2.5.0 (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached requests-2.31.0-py3-none-any.whl.metadata (4.6 kB)
Requirement already satisfied: packaging in /tmp/chex-env/lib/python3.10/site-packages (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1)) (23.2)
Requirement already satisfied: setuptools>=18.5 in /tmp/chex-env/lib/python3.10/site-packages (from IPython==7.16.3->-r requirements/requirements-docs.txt (line 5)) (69.0.3)
Collecting jedi<=0.17.2,>=0.10 (from IPython==7.16.3->-r requirements/requirements-docs.txt (line 5))
  Using cached jedi-0.17.2-py2.py3-none-any.whl (1.4 MB)
Collecting decorator (from IPython==7.16.3->-r requirements/requirements-docs.txt (line 5))
  Using cached decorator-5.1.1-py3-none-any.whl (9.1 kB)
Collecting pickleshare (from IPython==7.16.3->-r requirements/requirements-docs.txt (line 5))
  Using cached pickleshare-0.7.5-py2.py3-none-any.whl (6.9 kB)
Collecting traitlets>=4.2 (from IPython==7.16.3->-r requirements/requirements-docs.txt (line 5))
  Using cached traitlets-5.14.0-py3-none-any.whl.metadata (10 kB)
Collecting prompt-toolkit!=3.0.0,!=3.0.1,<3.1.0,>=2.0.0 (from IPython==7.16.3->-r requirements/requirements-docs.txt (line 5))
  Using cached prompt_toolkit-3.0.43-py3-none-any.whl.metadata (6.5 kB)
Collecting backcall (from IPython==7.16.3->-r requirements/requirements-docs.txt (line 5))
  Using cached backcall-0.2.0-py2.py3-none-any.whl (11 kB)
Collecting pexpect (from IPython==7.16.3->-r requirements/requirements-docs.txt (line 5))
  Using cached pexpect-4.9.0-py2.py3-none-any.whl.metadata (2.5 kB)
Collecting jupyter-client (from ipykernel==5.3.4->-r requirements/requirements-docs.txt (line 6))
  Using cached jupyter_client-8.6.0-py3-none-any.whl.metadata (8.3 kB)
Collecting tornado>=4.2 (from ipykernel==5.3.4->-r requirements/requirements-docs.txt (line 6))
  Using cached tornado-6.4-cp38-abi3-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (2.5 kB)
Collecting ply (from pandoc==1.0.2->-r requirements/requirements-docs.txt (line 7))
  Using cached ply-3.11-py2.py3-none-any.whl (49 kB)
Collecting importlib-metadata (from myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached importlib_metadata-7.0.1-py3-none-any.whl.metadata (4.9 kB)
Collecting ipywidgets<8,>=7.0.0 (from myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached ipywidgets-7.8.1-py2.py3-none-any.whl.metadata (1.9 kB)
Collecting jupyter-cache~=0.4.1 (from myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyter_cache-0.4.3-py3-none-any.whl (31 kB)
Collecting jupyter-sphinx~=0.3.2 (from myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyter_sphinx-0.3.2-py3-none-any.whl (20 kB)
Collecting myst-parser~=0.15.2 (from myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached myst_parser-0.15.2-py3-none-any.whl (46 kB)
Collecting nbconvert<7,>=5.6 (from myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached nbconvert-6.5.4-py3-none-any.whl (563 kB)
Collecting nbformat~=5.0 (from myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached nbformat-5.9.2-py3-none-any.whl.metadata (3.4 kB)
Requirement already satisfied: pyyaml in /tmp/chex-env/lib/python3.10/site-packages (from myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8)) (6.0.1)
Collecting sphinx-togglebutton~=0.2.2 (from myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached sphinx_togglebutton-0.2.3-py3-none-any.whl (6.1 kB)
Collecting cycler>=0.10 (from matplotlib==3.5.0->-r requirements/requirements-docs.txt (line 10))
  Using cached cycler-0.12.1-py3-none-any.whl.metadata (3.8 kB)
Collecting fonttools>=4.22.0 (from matplotlib==3.5.0->-r requirements/requirements-docs.txt (line 10))
  Using cached fonttools-4.47.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (157 kB)
Collecting kiwisolver>=1.0.1 (from matplotlib==3.5.0->-r requirements/requirements-docs.txt (line 10))
  Using cached kiwisolver-1.4.5-cp310-cp310-manylinux_2_12_x86_64.manylinux2010_x86_64.whl.metadata (6.4 kB)
Requirement already satisfied: numpy>=1.17 in /tmp/chex-env/lib/python3.10/site-packages (from matplotlib==3.5.0->-r requirements/requirements-docs.txt (line 10)) (1.26.2)
Collecting pillow>=6.2.0 (from matplotlib==3.5.0->-r requirements/requirements-docs.txt (line 10))
  Using cached Pillow-10.1.0-cp310-cp310-manylinux_2_28_x86_64.whl.metadata (9.5 kB)
Requirement already satisfied: pyparsing>=2.2.1 in /tmp/chex-env/lib/python3.10/site-packages (from matplotlib==3.5.0->-r requirements/requirements-docs.txt (line 10)) (3.1.1)
Collecting python-dateutil>=2.7 (from matplotlib==3.5.0->-r requirements/requirements-docs.txt (line 10))
  Using cached python_dateutil-2.8.2-py2.py3-none-any.whl (247 kB)
Collecting setuptools-scm>=4 (from matplotlib==3.5.0->-r requirements/requirements-docs.txt (line 10))
  Using cached setuptools_scm-8.0.4-py3-none-any.whl.metadata (6.4 kB)
Collecting comm>=0.1.3 (from ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached comm-0.2.0-py3-none-any.whl.metadata (3.7 kB)
Collecting ipython-genutils~=0.2.0 (from ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached ipython_genutils-0.2.0-py2.py3-none-any.whl (26 kB)
Collecting widgetsnbextension~=3.6.6 (from ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached widgetsnbextension-3.6.6-py2.py3-none-any.whl.metadata (1.3 kB)
Collecting jupyterlab-widgets<3,>=1.0.0 (from ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyterlab_widgets-1.1.7-py3-none-any.whl.metadata (3.7 kB)
Collecting parso<0.8.0,>=0.7.0 (from jedi<=0.17.2,>=0.10->IPython==7.16.3->-r requirements/requirements-docs.txt (line 5))
  Using cached parso-0.7.1-py2.py3-none-any.whl (109 kB)
Requirement already satisfied: MarkupSafe>=2.0 in /tmp/chex-env/lib/python3.10/site-packages (from Jinja2>=2.3->sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1)) (2.1.3)
Requirement already satisfied: attrs in /tmp/chex-env/lib/python3.10/site-packages (from jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8)) (23.1.0)
Collecting nbdime (from jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached nbdime-4.0.1-py3-none-any.whl.metadata (9.5 kB)
Collecting nbclient<0.6,>=0.2 (from jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached nbclient-0.5.13-py3-none-any.whl (70 kB)
Collecting sqlalchemy<1.5,>=1.3.12 (from jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached SQLAlchemy-1.4.50-cp310-cp310-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (10.0 kB)
Collecting markdown-it-py<2.0.0,>=1.0.0 (from myst-parser~=0.15.2->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached markdown_it_py-1.1.0-py3-none-any.whl (83 kB)
Collecting mdit-py-plugins~=0.2.8 (from myst-parser~=0.15.2->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached mdit_py_plugins-0.2.8-py3-none-any.whl (41 kB)
Collecting lxml (from nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached lxml-4.9.4-cp310-cp310-manylinux_2_28_x86_64.whl.metadata (3.7 kB)
Collecting beautifulsoup4 (from nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached beautifulsoup4-4.12.2-py3-none-any.whl (142 kB)
Collecting bleach (from nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached bleach-6.1.0-py3-none-any.whl.metadata (30 kB)
Collecting defusedxml (from nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached defusedxml-0.7.1-py2.py3-none-any.whl (25 kB)
Collecting entrypoints>=0.2.2 (from nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached entrypoints-0.4-py3-none-any.whl (5.3 kB)
Collecting jupyter-core>=4.7 (from nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyter_core-5.5.1-py3-none-any.whl.metadata (3.4 kB)
Collecting jupyterlab-pygments (from nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyterlab_pygments-0.3.0-py3-none-any.whl.metadata (4.4 kB)
Collecting mistune<2,>=0.8.1 (from nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached mistune-0.8.4-py2.py3-none-any.whl (16 kB)
Collecting pandocfilters>=1.4.1 (from nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached pandocfilters-1.5.0-py2.py3-none-any.whl (8.7 kB)
Collecting tinycss2 (from nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached tinycss2-1.2.1-py3-none-any.whl (21 kB)
Collecting fastjsonschema (from nbformat~=5.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached fastjsonschema-2.19.0-py3-none-any.whl.metadata (2.0 kB)
Collecting jsonschema>=2.6 (from nbformat~=5.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jsonschema-4.20.0-py3-none-any.whl.metadata (8.1 kB)
Collecting wcwidth (from prompt-toolkit!=3.0.0,!=3.0.1,<3.1.0,>=2.0.0->IPython==7.16.3->-r requirements/requirements-docs.txt (line 5))
  Using cached wcwidth-0.2.12-py2.py3-none-any.whl.metadata (14 kB)
Collecting six>=1.5 (from python-dateutil>=2.7->matplotlib==3.5.0->-r requirements/requirements-docs.txt (line 10))
  Using cached six-1.16.0-py2.py3-none-any.whl (11 kB)
Collecting charset-normalizer<4,>=2 (from requests>=2.5.0->sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached charset_normalizer-3.3.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (33 kB)
Collecting idna<4,>=2.5 (from requests>=2.5.0->sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached idna-3.6-py3-none-any.whl.metadata (9.9 kB)
Collecting urllib3<3,>=1.21.1 (from requests>=2.5.0->sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached urllib3-2.1.0-py3-none-any.whl.metadata (6.4 kB)
Collecting certifi>=2017.4.17 (from requests>=2.5.0->sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached certifi-2023.11.17-py3-none-any.whl.metadata (2.2 kB)
Requirement already satisfied: typing-extensions in /tmp/chex-env/lib/python3.10/site-packages (from setuptools-scm>=4->matplotlib==3.5.0->-r requirements/requirements-docs.txt (line 10)) (4.9.0)
Requirement already satisfied: tomli>=1 in /tmp/chex-env/lib/python3.10/site-packages (from setuptools-scm>=4->matplotlib==3.5.0->-r requirements/requirements-docs.txt (line 10)) (2.0.1)
Requirement already satisfied: wheel in /tmp/chex-env/lib/python3.10/site-packages (from sphinx-togglebutton~=0.2.2->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8)) (0.42.0)
INFO: pip is looking at multiple versions of sphinxcontrib-htmlhelp to determine which version is compatible with other requirements. This could take a while.
Collecting sphinxcontrib-htmlhelp>=2.0.0 (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached sphinxcontrib_htmlhelp-2.0.3-py3-none-any.whl.metadata (2.2 kB)
  Using cached sphinxcontrib_htmlhelp-2.0.2-py3-none-any.whl.metadata (2.2 kB)
  Using cached sphinxcontrib_htmlhelp-2.0.1-py3-none-any.whl (99 kB)
INFO: pip is looking at multiple versions of sphinxcontrib-serializinghtml to determine which version is compatible with other requirements. This could take a while.
Collecting sphinxcontrib-serializinghtml>=1.1.5 (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached sphinxcontrib_serializinghtml-1.1.8-py3-none-any.whl.metadata (2.3 kB)
  Using cached sphinxcontrib_serializinghtml-1.1.7-py3-none-any.whl.metadata (2.3 kB)
  Using cached sphinxcontrib_serializinghtml-1.1.6-py3-none-any.whl.metadata (2.3 kB)
  Using cached sphinxcontrib_serializinghtml-1.1.5-py2.py3-none-any.whl (94 kB)
Collecting zipp>=0.5 (from importlib-metadata->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached zipp-3.17.0-py3-none-any.whl.metadata (3.7 kB)
Collecting pyzmq>=23.0 (from jupyter-client->ipykernel==5.3.4->-r requirements/requirements-docs.txt (line 6))
  Using cached pyzmq-25.1.2-cp310-cp310-manylinux_2_28_x86_64.whl.metadata (4.9 kB)
Collecting ptyprocess>=0.5 (from pexpect->IPython==7.16.3->-r requirements/requirements-docs.txt (line 5))
  Using cached ptyprocess-0.7.0-py2.py3-none-any.whl (13 kB)
INFO: pip is looking at multiple versions of sphinxcontrib-applehelp to determine which version is compatible with other requirements. This could take a while.
Collecting sphinxcontrib-applehelp (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached sphinxcontrib_applehelp-1.0.6-py3-none-any.whl.metadata (2.2 kB)
  Using cached sphinxcontrib_applehelp-1.0.5-py3-none-any.whl.metadata (2.2 kB)
  Using cached sphinxcontrib_applehelp-1.0.4-py3-none-any.whl (120 kB)
INFO: pip is looking at multiple versions of sphinxcontrib-devhelp to determine which version is compatible with other requirements. This could take a while.
Collecting sphinxcontrib-devhelp (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached sphinxcontrib_devhelp-1.0.4-py3-none-any.whl.metadata (2.2 kB)
  Using cached sphinxcontrib_devhelp-1.0.3-py3-none-any.whl.metadata (2.2 kB)
  Using cached sphinxcontrib_devhelp-1.0.2-py2.py3-none-any.whl (84 kB)
INFO: pip is looking at multiple versions of sphinxcontrib-qthelp to determine which version is compatible with other requirements. This could take a while.
Collecting sphinxcontrib-qthelp (from sphinx==4.5.0->-r requirements/requirements-docs.txt (line 1))
  Using cached sphinxcontrib_qthelp-1.0.5-py3-none-any.whl.metadata (2.2 kB)
  Using cached sphinxcontrib_qthelp-1.0.4-py3-none-any.whl.metadata (2.2 kB)
  Using cached sphinxcontrib_qthelp-1.0.3-py2.py3-none-any.whl (90 kB)
Collecting jsonschema-specifications>=2023.03.6 (from jsonschema>=2.6->nbformat~=5.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jsonschema_specifications-2023.11.2-py3-none-any.whl.metadata (3.0 kB)
Collecting referencing>=0.28.4 (from jsonschema>=2.6->nbformat~=5.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached referencing-0.32.0-py3-none-any.whl.metadata (2.7 kB)
Collecting rpds-py>=0.7.1 (from jsonschema>=2.6->nbformat~=5.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached rpds_py-0.15.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (4.1 kB)
Requirement already satisfied: platformdirs>=2.5 in /tmp/chex-env/lib/python3.10/site-packages (from jupyter-core>=4.7->nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8)) (4.1.0)
INFO: pip is looking at multiple versions of markdown-it-py to determine which version is compatible with other requirements. This could take a while.
Collecting markdown-it-py<2.0.0,>=1.0.0 (from myst-parser~=0.15.2->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached markdown_it_py-1.0.0-py3-none-any.whl (83 kB)
Collecting jsonschema>=2.6 (from nbformat~=5.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jsonschema-4.19.2-py3-none-any.whl.metadata (7.9 kB)
  Using cached jsonschema-4.19.1-py3-none-any.whl.metadata (7.9 kB)
  Using cached jsonschema-4.19.0-py3-none-any.whl.metadata (8.2 kB)
INFO: pip is still looking at multiple versions of markdown-it-py to determine which version is compatible with other requirements. This could take a while.
  Using cached jsonschema-4.18.6-py3-none-any.whl.metadata (7.8 kB)
  Using cached jsonschema-4.18.5-py3-none-any.whl.metadata (7.7 kB)
  Using cached jsonschema-4.18.4-py3-none-any.whl.metadata (7.8 kB)
INFO: This is taking longer than usual. You might need to provide the dependency resolver with stricter constraints to reduce runtime. See https://pip.pypa.io/warnings/backtracking for guidance. If you want to abort this run, press Ctrl + C.
  Using cached jsonschema-4.18.3-py3-none-any.whl.metadata (7.9 kB)
  Using cached jsonschema-4.18.2-py3-none-any.whl.metadata (7.8 kB)
  Using cached jsonschema-4.18.1-py3-none-any.whl.metadata (7.8 kB)
  Using cached jsonschema-4.18.0-py3-none-any.whl.metadata (10 kB)
  Using cached jsonschema-4.17.3-py3-none-any.whl (90 kB)
Collecting pyrsistent!=0.17.0,!=0.17.1,!=0.17.2,>=0.14.0 (from jsonschema>=2.6->nbformat~=5.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached pyrsistent-0.20.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (27 kB)
Collecting attrs (from jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached attrs-21.4.0-py2.py3-none-any.whl (60 kB)
Collecting nest-asyncio (from nbclient<0.6,>=0.2->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached nest_asyncio-1.5.8-py3-none-any.whl.metadata (2.8 kB)
Collecting greenlet!=0.4.17 (from sqlalchemy<1.5,>=1.3.12->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached greenlet-3.0.3-cp310-cp310-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl.metadata (3.8 kB)
Collecting notebook>=4.4.1 (from widgetsnbextension~=3.6.6->ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached notebook-7.0.6-py3-none-any.whl.metadata (10 kB)
Collecting soupsieve>1.2 (from beautifulsoup4->nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached soupsieve-2.5-py3-none-any.whl.metadata (4.7 kB)
Collecting webencodings (from bleach->nbconvert<7,>=5.6->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached webencodings-0.5.1-py2.py3-none-any.whl (11 kB)
Collecting colorama (from nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached colorama-0.4.6-py2.py3-none-any.whl (25 kB)
Collecting gitpython!=2.1.4,!=2.1.5,!=2.1.6 (from nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached GitPython-3.1.40-py3-none-any.whl.metadata (12 kB)
Collecting jupyter-server (from nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyter_server-2.12.1-py3-none-any.whl.metadata (8.4 kB)
Collecting jupyter-server-mathjax>=0.2.2 (from nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyter_server_mathjax-0.2.6-py3-none-any.whl (3.1 MB)
Collecting gitdb<5,>=4.0.1 (from gitpython!=2.1.4,!=2.1.5,!=2.1.6->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached gitdb-4.0.11-py3-none-any.whl.metadata (1.2 kB)
Collecting anyio>=3.1.0 (from jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached anyio-4.2.0-py3-none-any.whl.metadata (4.6 kB)
Collecting argon2-cffi (from jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached argon2_cffi-23.1.0-py3-none-any.whl.metadata (5.2 kB)
Collecting jupyter-events>=0.9.0 (from jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyter_events-0.9.0-py3-none-any.whl.metadata (5.7 kB)
Collecting jupyter-server-terminals (from jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyter_server_terminals-0.5.0-py3-none-any.whl.metadata (5.6 kB)
Collecting overrides (from jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached overrides-7.4.0-py3-none-any.whl.metadata (5.7 kB)
Collecting prometheus-client (from jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached prometheus_client-0.19.0-py3-none-any.whl.metadata (1.8 kB)
Collecting send2trash>=1.8.2 (from jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached Send2Trash-1.8.2-py3-none-any.whl (18 kB)
Collecting terminado>=0.8.3 (from jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached terminado-0.18.0-py3-none-any.whl.metadata (5.8 kB)
Collecting websocket-client (from jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached websocket_client-1.7.0-py3-none-any.whl.metadata (7.9 kB)
Collecting jupyterlab-server<3,>=2.22.1 (from notebook>=4.4.1->widgetsnbextension~=3.6.6->ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyterlab_server-2.25.2-py3-none-any.whl.metadata (5.9 kB)
Collecting jupyterlab<5,>=4.0.2 (from notebook>=4.4.1->widgetsnbextension~=3.6.6->ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyterlab-4.0.9-py3-none-any.whl.metadata (15 kB)
Collecting notebook-shim<0.3,>=0.2 (from notebook>=4.4.1->widgetsnbextension~=3.6.6->ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached notebook_shim-0.2.3-py3-none-any.whl (13 kB)
Collecting sniffio>=1.1 (from anyio>=3.1.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached sniffio-1.3.0-py3-none-any.whl (10 kB)
Requirement already satisfied: exceptiongroup>=1.0.2 in /tmp/chex-env/lib/python3.10/site-packages (from anyio>=3.1.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8)) (1.2.0)
Collecting smmap<6,>=3.0.1 (from gitdb<5,>=4.0.1->gitpython!=2.1.4,!=2.1.5,!=2.1.6->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached smmap-5.0.1-py3-none-any.whl.metadata (4.3 kB)
INFO: pip is looking at multiple versions of jupyter-events to determine which version is compatible with other requirements. This could take a while.
Collecting jupyter-server (from nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyter_server-2.12.0-py3-none-any.whl.metadata (8.4 kB)
  Using cached jupyter_server-2.11.2-py3-none-any.whl.metadata (8.4 kB)
  Using cached jupyter_server-2.11.1-py3-none-any.whl.metadata (8.4 kB)
  Using cached jupyter_server-2.10.1-py3-none-any.whl.metadata (8.4 kB)
  Using cached jupyter_server-2.10.0-py3-none-any.whl.metadata (8.4 kB)
Collecting jupyter-events>=0.6.0 (from jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyter_events-0.8.0-py3-none-any.whl.metadata (5.7 kB)
  Using cached jupyter_events-0.7.0-py3-none-any.whl.metadata (5.5 kB)
INFO: pip is still looking at multiple versions of jupyter-events to determine which version is compatible with other requirements. This could take a while.
  Using cached jupyter_events-0.6.3-py3-none-any.whl (18 kB)
Collecting python-json-logger>=2.0.4 (from jupyter-events>=0.6.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached python_json_logger-2.0.7-py3-none-any.whl (8.1 kB)
Collecting rfc3339-validator (from jupyter-events>=0.6.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached rfc3339_validator-0.1.4-py2.py3-none-any.whl (3.5 kB)
Collecting rfc3986-validator>=0.1.1 (from jupyter-events>=0.6.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached rfc3986_validator-0.1.1-py2.py3-none-any.whl (4.2 kB)
Collecting async-lru>=1.0.0 (from jupyterlab<5,>=4.0.2->notebook>=4.4.1->widgetsnbextension~=3.6.6->ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached async_lru-2.0.4-py3-none-any.whl.metadata (4.5 kB)
Collecting jupyter-lsp>=2.0.0 (from jupyterlab<5,>=4.0.2->notebook>=4.4.1->widgetsnbextension~=3.6.6->ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyter_lsp-2.2.1-py3-none-any.whl.metadata (1.8 kB)
Collecting json5>=0.9.0 (from jupyterlab-server<3,>=2.22.1->notebook>=4.4.1->widgetsnbextension~=3.6.6->ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached json5-0.9.14-py2.py3-none-any.whl.metadata (10 kB)
INFO: pip is looking at multiple versions of jupyterlab-server to determine which version is compatible with other requirements. This could take a while.
Collecting jupyterlab-server<3,>=2.22.1 (from notebook>=4.4.1->widgetsnbextension~=3.6.6->ipywidgets<8,>=7.0.0->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jupyterlab_server-2.25.1-py3-none-any.whl.metadata (5.9 kB)
  Using cached jupyterlab_server-2.25.0-py3-none-any.whl.metadata (5.9 kB)
  Using cached jupyterlab_server-2.24.0-py3-none-any.whl.metadata (5.8 kB)
Collecting argon2-cffi-bindings (from argon2-cffi->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached argon2_cffi_bindings-21.2.0-cp36-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (86 kB)
INFO: pip is looking at multiple versions of jsonschema[format-nongpl] to determine which version is compatible with other requirements. This could take a while.
INFO: pip is still looking at multiple versions of jsonschema[format-nongpl] to determine which version is compatible with other requirements. This could take a while.
Collecting fqdn (from jsonschema[format-nongpl]>=3.2.0->jupyter-events>=0.6.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached fqdn-1.5.1-py3-none-any.whl (9.1 kB)
Collecting isoduration (from jsonschema[format-nongpl]>=3.2.0->jupyter-events>=0.6.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached isoduration-20.11.0-py3-none-any.whl (11 kB)
Collecting jsonpointer>1.13 (from jsonschema[format-nongpl]>=3.2.0->jupyter-events>=0.6.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached jsonpointer-2.4-py2.py3-none-any.whl.metadata (2.5 kB)
Collecting uri-template (from jsonschema[format-nongpl]>=3.2.0->jupyter-events>=0.6.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached uri_template-1.3.0-py3-none-any.whl.metadata (8.8 kB)
Collecting webcolors>=1.11 (from jsonschema[format-nongpl]>=3.2.0->jupyter-events>=0.6.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached webcolors-1.13-py3-none-any.whl (14 kB)
Collecting cffi>=1.0.1 (from argon2-cffi-bindings->argon2-cffi->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached cffi-1.16.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (1.5 kB)
Collecting pycparser (from cffi>=1.0.1->argon2-cffi-bindings->argon2-cffi->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached pycparser-2.21-py2.py3-none-any.whl (118 kB)
Collecting arrow>=0.15.0 (from isoduration->jsonschema[format-nongpl]>=3.2.0->jupyter-events>=0.6.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached arrow-1.3.0-py3-none-any.whl.metadata (7.5 kB)
Collecting types-python-dateutil>=2.8.10 (from arrow>=0.15.0->isoduration->jsonschema[format-nongpl]>=3.2.0->jupyter-events>=0.6.0->jupyter-server->nbdime->jupyter-cache~=0.4.1->myst_nb==0.13.1->-r requirements/requirements-docs.txt (line 8))
  Using cached types_python_dateutil-2.8.19.14-py3-none-any.whl.metadata (1.6 kB)
Using cached Babel-2.14.0-py3-none-any.whl (11.0 MB)
Using cached cycler-0.12.1-py3-none-any.whl (8.3 kB)
Using cached fonttools-4.47.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.6 MB)
Using cached ipywidgets-7.8.1-py2.py3-none-any.whl (124 kB)
Using cached kiwisolver-1.4.5-cp310-cp310-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (1.6 MB)
Using cached nbformat-5.9.2-py3-none-any.whl (77 kB)
Using cached Pillow-10.1.0-cp310-cp310-manylinux_2_28_x86_64.whl (3.6 MB)
Using cached prompt_toolkit-3.0.43-py3-none-any.whl (386 kB)
Using cached pygments-2.17.2-py3-none-any.whl (1.2 MB)
Using cached requests-2.31.0-py3-none-any.whl (62 kB)
Using cached setuptools_scm-8.0.4-py3-none-any.whl (42 kB)
Using cached tornado-6.4-cp38-abi3-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl (435 kB)
Using cached traitlets-5.14.0-py3-none-any.whl (85 kB)
Using cached importlib_metadata-7.0.1-py3-none-any.whl (23 kB)
Using cached jupyter_client-8.6.0-py3-none-any.whl (105 kB)
Using cached pexpect-4.9.0-py2.py3-none-any.whl (63 kB)
Using cached certifi-2023.11.17-py3-none-any.whl (162 kB)
Using cached charset_normalizer-3.3.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (142 kB)
Using cached comm-0.2.0-py3-none-any.whl (7.0 kB)
Using cached idna-3.6-py3-none-any.whl (61 kB)
Using cached jupyter_core-5.5.1-py3-none-any.whl (28 kB)
Using cached jupyterlab_widgets-1.1.7-py3-none-any.whl (295 kB)
Using cached pyzmq-25.1.2-cp310-cp310-manylinux_2_28_x86_64.whl (1.1 MB)
Using cached SQLAlchemy-1.4.50-cp310-cp310-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.6 MB)
Using cached urllib3-2.1.0-py3-none-any.whl (104 kB)
Using cached widgetsnbextension-3.6.6-py2.py3-none-any.whl (1.6 MB)
Using cached zipp-3.17.0-py3-none-any.whl (7.4 kB)
Using cached bleach-6.1.0-py3-none-any.whl (162 kB)
Using cached fastjsonschema-2.19.0-py3-none-any.whl (23 kB)
Using cached jupyterlab_pygments-0.3.0-py3-none-any.whl (15 kB)
Using cached lxml-4.9.4-cp310-cp310-manylinux_2_28_x86_64.whl (7.9 MB)
Using cached nbdime-4.0.1-py3-none-any.whl (5.9 MB)
Using cached wcwidth-0.2.12-py2.py3-none-any.whl (34 kB)
Using cached GitPython-3.1.40-py3-none-any.whl (190 kB)
Using cached greenlet-3.0.3-cp310-cp310-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl (616 kB)
Using cached jupyter_server-2.10.0-py3-none-any.whl (377 kB)
Using cached notebook-7.0.6-py3-none-any.whl (4.0 MB)
Using cached pyrsistent-0.20.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (117 kB)
Using cached soupsieve-2.5-py3-none-any.whl (36 kB)
Using cached nest_asyncio-1.5.8-py3-none-any.whl (5.3 kB)
Using cached anyio-4.2.0-py3-none-any.whl (85 kB)
Using cached gitdb-4.0.11-py3-none-any.whl (62 kB)
Using cached jupyterlab-4.0.9-py3-none-any.whl (9.2 MB)
Using cached jupyterlab_server-2.24.0-py3-none-any.whl (57 kB)
Using cached terminado-0.18.0-py3-none-any.whl (14 kB)
Using cached argon2_cffi-23.1.0-py3-none-any.whl (15 kB)
Using cached jupyter_server_terminals-0.5.0-py3-none-any.whl (13 kB)
Using cached overrides-7.4.0-py3-none-any.whl (17 kB)
Using cached prometheus_client-0.19.0-py3-none-any.whl (54 kB)
Using cached websocket_client-1.7.0-py3-none-any.whl (58 kB)
Using cached async_lru-2.0.4-py3-none-any.whl (6.1 kB)
Using cached json5-0.9.14-py2.py3-none-any.whl (19 kB)
Using cached jupyter_lsp-2.2.1-py3-none-any.whl (66 kB)
Using cached smmap-5.0.1-py3-none-any.whl (24 kB)
Using cached cffi-1.16.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (443 kB)
Using cached jsonpointer-2.4-py2.py3-none-any.whl (7.8 kB)
Using cached uri_template-1.3.0-py3-none-any.whl (11 kB)
Using cached arrow-1.3.0-py3-none-any.whl (66 kB)
Using cached types_python_dateutil-2.8.19.14-py3-none-any.whl (9.4 kB)
Installing collected packages: webencodings, wcwidth, types-python-dateutil, snowballstemmer, ptyprocess, ply, pickleshare, mistune, json5, ipython-genutils, fastjsonschema, backcall, zipp, websocket-client, webcolors, urllib3, uri-template, traitlets, tornado, tinycss2, sphinxcontrib-serializinghtml, sphinxcontrib-qthelp, sphinxcontrib-jsmath, sphinxcontrib-htmlhelp, sphinxcontrib-devhelp, sphinxcontrib-applehelp, soupsieve, sniffio, smmap, six, setuptools-scm, send2trash, rfc3986-validator, pyzmq, python-json-logger, pyrsistent, Pygments, pycparser, prompt-toolkit, prometheus-client, pillow, pexpect, parso, pandocfilters, pandoc, overrides, nest-asyncio, lxml, kiwisolver, jupyterlab-widgets, jupyterlab-pygments, jsonpointer, imagesize, idna, greenlet, fqdn, fonttools, entrypoints, docutils, defusedxml, decorator, cycler, colorama, charset-normalizer, certifi, babel, attrs, async-lru, alabaster, terminado, sqlalchemy, rfc3339-validator, requests, python-dateutil, markdown-it-py, jupyter-core, jsonschema, jedi, importlib-metadata, gitdb, comm, cffi, bleach, beautifulsoup4, anyio, sphinx, nbformat, mdit-py-plugins, matplotlib, jupyter-server-terminals, jupyter-client, IPython, gitpython, arrow, argon2-cffi-bindings, sphinxcontrib-katex, sphinx-togglebutton, sphinx_rtd_theme, sphinx-autodoc-typehints, nbclient, myst-parser, isoduration, ipykernel, argon2-cffi, nbconvert, jupyter-events, jupyter-server, notebook-shim, jupyterlab-server, jupyter-server-mathjax, jupyter-lsp, nbdime, jupyterlab, notebook, jupyter-cache, widgetsnbextension, ipywidgets, jupyter-sphinx, myst_nb
  Attempting uninstall: attrs
    Found existing installation: attrs 23.1.0
    Uninstalling attrs-23.1.0:
      Successfully uninstalled attrs-23.1.0
Successfully installed IPython-7.16.3 Pygments-2.17.2 alabaster-0.7.13 anyio-4.2.0 argon2-cffi-23.1.0 argon2-cffi-bindings-21.2.0 arrow-1.3.0 async-lru-2.0.4 attrs-21.4.0 babel-2.14.0 backcall-0.2.0 beautifulsoup4-4.12.2 bleach-6.1.0 certifi-2023.11.17 cffi-1.16.0 charset-normalizer-3.3.2 colorama-0.4.6 comm-0.2.0 cycler-0.12.1 decorator-5.1.1 defusedxml-0.7.1 docutils-0.16 entrypoints-0.4 fastjsonschema-2.19.0 fonttools-4.47.0 fqdn-1.5.1 gitdb-4.0.11 gitpython-3.1.40 greenlet-3.0.3 idna-3.6 imagesize-1.4.1 importlib-metadata-7.0.1 ipykernel-5.3.4 ipython-genutils-0.2.0 ipywidgets-7.8.1 isoduration-20.11.0 jedi-0.17.2 json5-0.9.14 jsonpointer-2.4 jsonschema-4.17.3 jupyter-cache-0.4.3 jupyter-client-8.6.0 jupyter-core-5.5.1 jupyter-events-0.6.3 jupyter-lsp-2.2.1 jupyter-server-2.10.0 jupyter-server-mathjax-0.2.6 jupyter-server-terminals-0.5.0 jupyter-sphinx-0.3.2 jupyterlab-4.0.9 jupyterlab-pygments-0.3.0 jupyterlab-server-2.24.0 jupyterlab-widgets-1.1.7 kiwisolver-1.4.5 lxml-4.9.4 markdown-it-py-1.1.0 matplotlib-3.5.0 mdit-py-plugins-0.2.8 mistune-0.8.4 myst-parser-0.15.2 myst_nb-0.13.1 nbclient-0.5.13 nbconvert-6.5.4 nbdime-4.0.1 nbformat-5.9.2 nest-asyncio-1.5.8 notebook-7.0.6 notebook-shim-0.2.3 overrides-7.4.0 pandoc-1.0.2 pandocfilters-1.5.0 parso-0.7.1 pexpect-4.9.0 pickleshare-0.7.5 pillow-10.1.0 ply-3.11 prometheus-client-0.19.0 prompt-toolkit-3.0.43 ptyprocess-0.7.0 pycparser-2.21 pyrsistent-0.20.0 python-dateutil-2.8.2 python-json-logger-2.0.7 pyzmq-25.1.2 requests-2.31.0 rfc3339-validator-0.1.4 rfc3986-validator-0.1.1 send2trash-1.8.2 setuptools-scm-8.0.4 six-1.16.0 smmap-5.0.1 sniffio-1.3.0 snowballstemmer-2.2.0 soupsieve-2.5 sphinx-4.5.0 sphinx-autodoc-typehints-1.11.1 sphinx-togglebutton-0.2.3 sphinx_rtd_theme-1.0.0 sphinxcontrib-applehelp-1.0.4 sphinxcontrib-devhelp-1.0.2 sphinxcontrib-htmlhelp-2.0.1 sphinxcontrib-jsmath-1.0.1 sphinxcontrib-katex-0.8.6 sphinxcontrib-qthelp-1.0.3 sphinxcontrib-serializinghtml-1.1.5 sqlalchemy-1.4.50 terminado-0.18.0 tinycss2-1.2.1 tornado-6.4 traitlets-5.14.0 types-python-dateutil-2.8.19.14 uri-template-1.3.0 urllib3-2.1.0 wcwidth-0.2.12 webcolors-1.13 webencodings-0.5.1 websocket-client-1.7.0 widgetsnbextension-3.6.6 zipp-3.17.0
Running Sphinx v4.5.0
making output directory... done
[autosummary] generating autosummary for: api.rst, index.rst
loading intersphinx inventory from https://jax.readthedocs.io/en/latest/objects.inv...
building [mo]: targets for 0 po files that are out of date
building [coverage_check]: coverage_check
updating environment: [new config] 2 added, 0 changed, 0 removed
reading sources... [ 50%] api
reading sources... [100%] index

looking for now-outdated files... none found
pickling environment... done
checking consistency... done
build succeeded.
Running Sphinx v4.5.0
making output directory... done
loading pickled environment... done
[autosummary] generating autosummary for: api.rst, index.rst
building [mo]: targets for 0 po files that are out of date
building [html]: targets for 2 source files that are out of date
updating environment: 0 added, 0 changed, 0 removed
looking for now-outdated files... none found
preparing documents... done
writing output... [ 50%] api
writing output... [100%] index

generating indices... genindex done
writing additional pages... search done
copying static files... done
copying extra files... done
dumping search index in English (code: en)... done
dumping object inventory... done
build succeeded.

The HTML pages are in _build/html.
All tests passed. Congrats!