code: def setUp(self):
      self.one_hot_targets = jnp.array(
          [[1, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 0]]
      )
      self.logits = jnp.array(
          [
              [0.41, 0.39, 0.2],
              [0.4, 0.6, 0],
              [0.5, 0.1, 0.4],
              [0.3, 0.5, 0.2],
          ]
      )
      super().setUp()
  
file_path: scenic/scenic/projects/svvit/tests/metrics_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import jax.numpy
  from scenic.projects.svvit import metrics
label_desc: Use image node_id 1 for calling the MetricsTest obj's underlying member method code with example usage: obj.setUp() without return types