code: def test_transform_to_2hot(self):
      y = transforms.transform_to_2hot(
          scalar=jnp.array(TWO_HOT_SCALARS),
          min_value=-1.0,
          max_value=1.0,
          num_bins=TWO_HOT_BINS,
      )
  
      np.testing.assert_allclose(
          y, np.array(TWO_HOT_PROBABILITIES), atol=1e-4
      )
  
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import jax
  import jax.numpy
  import numpy
  from rlax._src import transforms
label_desc: Use image node_id 14 for calling the TransformsTest obj's underlying member method code with example usage: obj.test_transform_to_2hot() without return types