code: def _apply_update(u, p):
      if u is None:
          return p
      else:
          return p + u
  
imports: import jax.tree_util
  from jaxtyping import PyTree
label_desc: Use image node_id 1 for calling a global function with example usage: _apply_update(u, p) and returns: p, unknown