code: def _unchunk_array_leaves_in_place(d):
      """Convert chunked array leaves back into array leaves, in place."""
      if isinstance(d, dict):
          if "__msgpack_chunked_array__" in d:
              return _unchunk(d)
          else:
              for k, v in d.items():
                  if (
                      isinstance(v, dict)
                      and "__msgpack_chunked_array__" in v
                  ):
                      d[k] = _unchunk(v)
                  elif isinstance(v, dict):
                      _unchunk_array_leaves_in_place(v)
      return d
  
imports: import enum
  import threading
  from contextlib import contextmanager
  from typing import Any, Dict, List
  import jax
  import msgpack
  import numpy
label_desc: Use image node_id 22 for calling a global function with example usage: _unchunk_array_leaves_in_place(d) and returns: d, _unchunk