code: def test_sigmoid_vector(self):
      sigmoid = self.variant(transforms.sigmoid)
      logit = self.variant(transforms.logit)
      # Test output.
      np.testing.assert_allclose(
          logit(sigmoid(self.xs)), self.xs, atol=1e-3
      )
  
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 5 for calling the TransformsTest obj's underlying member method code with example usage: obj.test_sigmoid_vector() without return types