code: def test_expectile_loss_vector(
      self, expectile, predictions, expected_loss
  ):
      expectile_loss = self.variant(losses.expectile_loss)
      predictions = jnp.array(predictions)
      expected_loss = jnp.array(expected_loss)
      targets = jnp.zeros_like(predictions)
      # Test output.
      np.testing.assert_allclose(
          expectile_loss(
              predictions=predictions,
              targets=targets,
              expectile=expectile,
          ),
          expected_loss,
          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 losses
label_desc: Use image node_id 1 for calling the ExpectileLossTest obj's underlying member method code with example usage: obj.test_expectile_loss_vector(expectile, predictions, expected_loss) without return types