docstr: Tests for a batch, cumulant_type='decrease'.
docstr_tok: ["Tests","for","a","batch",",","cumulant_type='decrease","'","."]
code: def test_feature_control_rewards_decrease_batch(self):
      """Tests for a batch, cumulant_type='decrease'."""
      feature_control_rewards = self.variant(
          jax.vmap(
              functools.partial(
                  general_value_functions.feature_control_rewards,
                  cumulant_type="decrease",
              )
          )
      )
      # Test pseudo rewards.
      rs = feature_control_rewards(self.features)
      np.testing.assert_allclose(self.exp_decrease, rs, rtol=1e-5)
  
file_path: rlax/rlax/_src/general_value_functions_test.py
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import jax
  import numpy
  from rlax._src import general_value_functions
label_desc: Use image node_id 5 for calling the FeatureControlTest obj's underlying member method code with example usage: obj.test_feature_control_rewards_decrease_batch() without return types