code: def transform_values(build_targets, *value_argnums):
      """Decorator to convert targets to use transformed value function."""
  
      @functools.wraps(build_targets)
      def wrapped_build_targets(tx_pair, *args, **kwargs):
          tx_args = list(args)
          for index in value_argnums:
              tx_args[index] = tx_pair.apply_inv(tx_args[index])
  
          targets = build_targets(*tx_args, **kwargs)
          return tx_pair.apply(targets)
  
      return wrapped_build_targets
  
file_path: rlax/rlax/_src/nonlinear_bellman.py
imports: import collections
  import functools
  import chex
  import jax.numpy
  from rlax._src import base
  from rlax._src import multistep
  from rlax._src import transforms
label_desc: Use image node_id 5 for calling a global function with example usage: transform_values(build_targets) and returns: wrapped_build_targets, tx_pair