code: def test_weighted_top_one_correctly_classified(
      self, weights, expected_correct
  ):
      logits = np.array(
          [
              [[1.2, -0.2, 0.1], [0.2, -0.4, -0.4], [-1.0, -1.0, -1.0]],
              [[-0.4, 0.5, -1.0], [2.0, 1.0, 3.0], [-1.0, -1.0, -1.0]],
          ],
          dtype=np.float32,
      )
      multihot_targets = np.array(
          [
              [[1, 0, 0], [0, 0, 0], [0, 0, 0]],
              [[0, 0, 0], [0, 1, 0], [0, 0, 0]],
          ],
          dtype=np.int32,
      )
      correct = temporal_localization_base_model.weighted_top_one_correctly_classified(  # pylint: disable=line-too-long
          logits, multihot_targets, weights=weights
      )
      np.testing.assert_equal(correct, expected_correct)
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from flax import jax_utils
  import jax
  import ml_collections
  import numpy
  from scenic.projects.unloc import temporal_localization_base_model
label_desc: Use image node_id 1 for calling the TemporalLocalizationBaseModelTest obj's underlying member method code with example usage: obj.test_weighted_top_one_correctly_classified(weights, expected_correct) without return types