code: def check_callable(fun):
      # In Python 3.10+, the only thing stopping us from supporting staticmethods
      # is that we can't take weak references to them, which the C++ JIT requires.
      if isinstance(fun, staticmethod):
          raise TypeError(
              f"staticmethod arguments are not supported, got {fun}"
          )
      if not callable(fun):
          raise TypeError(f"Expected a callable value, got {fun}")
      if inspect.isgeneratorfunction(fun):
          raise TypeError(
              f"Expected a function, got a generator function: {fun}"
          )
  
imports: from __future__ import annotations
  from collections.abc import Iterable, Sequence
  import inspect
  import operator
  from functools import partial
  from typing import Any, Callable
  import warnings
  import numpy
  from jax._src import core
  from jax._src import dtypes
  from jax._src.abstract_arrays import numpy_scalar_types
  from jax._src.core import ShapedArray
  from jax._src.tree_util import PyTreeDef, tree_flatten, tree_unflatten, tree_map, tree_structure, treedef_children, generate_key_paths, keystr, broadcast_prefix, prefix_errors
  from jax._src.tree_util import _replace_nones
  from jax._src import linear_util
  from jax._src.linear_util import TracingDebugInfo
  from jax._src.util import safe_map, WrapKwArgs, Hashable, HashableFunction, Unhashable
  from jax._src import traceback_util
label_desc: Use image node_id 25 for calling a global function with example usage: check_callable(fun) without return types