code: def _equal_code(fn1: Optional[Callable], fn2: Optional[Callable]):
      """Checks whether fn1 and fn2 both have the same code.
  
      It's essentially impossible to see if two functions are equivalent, so this won't,
      and isn't intended, to catch every possible difference between fn1 and fn2. But it
      should at least catch the common case that `is_leaf` is specified for one input and
      not specified for the other.
      """
      sentinel1 = object()
      sentinel2 = object()
      code1 = getattr(
          getattr(fn1, "__code__", sentinel1), "co_code", sentinel2
      )
      code2 = getattr(
          getattr(fn2, "__code__", sentinel1), "co_code", sentinel2
      )
      return type(code1) == type(code2) and code1 == code2
  
imports: import operator
  from collections.abc import Callable
  from typing import Any, Optional, TYPE_CHECKING
  import jax
  import jax.tree_util
label_desc: Use image node_id 1 for calling a global function with example usage: _equal_code(fn1, fn2) and returns: unknown