code: def _cond_wrapper(t_fn, f_fn, scope, pred, *ops, variables, rngs):
      return lift.cond(
          pred, t_fn, f_fn, scope, *ops, variables=variables, rngs=rngs
      )
  
imports: import dataclasses
  import functools
  import inspect
  from typing import Any, Callable, Dict, Iterable, Mapping, Optional, Sequence, Tuple, Type, TypeVar, Union
  from flax import errors, struct, traceback_util
  from flax.core import Scope, lift, meta
  from flax.core.frozen_dict import FrozenDict
  from flax.linen import module
  from flax.linen.module import Module, Variable, _derive_profiling_name, _get_unbound_fn, wrap_method_once
  import jax
label_desc: Use image node_id 22 for calling a global function with example usage: _cond_wrapper(t_fn, f_fn, scope, pred) and returns: lift