code: def doc_remove_args(*args):
      def doc_remove_args_impl(fn):
          if getattr(typing, "GENERATING_DOCUMENTATION", False):
              sig = inspect.signature(fn)
              new_params = []
              for param in sig.parameters.values():
                  if param.name not in args:
                      new_params.append(param)
              sig = sig.replace(parameters=new_params)
              # Force-set it when doing `doc_remove_args(filter_jit(...))`.
              object.__setattr__(fn, "__signature__", sig)
          return fn
  
      return doc_remove_args_impl
  
file_path: equinox/equinox/_doc_utils.py
imports: import inspect
  import typing
  from typing import TYPE_CHECKING, TypeVar
label_desc: Use image node_id 2 for calling a global function with example usage: doc_remove_args() and returns: doc_remove_args_impl, fn