code: def _fixup_closure(leaf):
      if isinstance(leaf, types.FunctionType):
          return _Closure(leaf)
      else:
          return leaf
  
file_path: equinox/equinox/internal/_closure_to_pytree.py
imports: import inspect
  import types
  from typing import Any, Optional
  import jax.tree_util
  from .._module import Module
label_desc: Use image node_id 3 for calling a global function with example usage: _fixup_closure(leaf) and returns: _Closure, leaf