code: def __call__(self, **kwargs: tp.Any):
      current_flags = FLAGS_CONTEXT.flags_stack[-1]
      FLAGS_CONTEXT.flags_stack.append(
          MappingProxyType(dict(current_flags, **kwargs))
      )
      try:
          yield
      finally:
          FLAGS_CONTEXT.flags_stack.pop()
  
file_path: flax/flax/experimental/nnx/nnx/flaglib.py
imports: from __future__ import annotations
  import dataclasses
  import threading
  import typing
  from contextlib import contextmanager
  from types import MappingProxyType
label_desc: Use image node_id 5 for calling the Flags obj's underlying member method code with example usage: obj.__call__() without return types