code: def _mutable(obj: P) -> tp.Iterator[None]:
      vars(obj)["_pytree__is_mutable"] = True
      try:
          yield
      finally:
          del vars(obj)["_pytree__is_mutable"]
  
imports: import contextlib
  import dataclasses
  import importlib.util
  import inspect
  import typing
  from abc import ABCMeta
  from copy import copy
  from functools import partial
  from types import MappingProxyType
  import jax
  import numpy
  from flax.experimental.nnx.nnx import module
  from flax.experimental.nnx.nnx import reprlib, variables
  from flax.experimental.nnx.nnx.state import State
label_desc: Use image node_id 1 for calling a global function with example usage: _mutable(obj) without return types