code: def test_weighted_unnormalized_iou_loss(self, weights, loss_type):
      batch_size, num_frames, num_classes = 2, 8, 10
      displacements = np.ones(
          (batch_size, num_frames, num_classes, 2), dtype=np.float32
      )
      gt_displacements = np.ones(
          (batch_size, num_frames, num_classes, 2), dtype=np.float32
      )
      label = np.zeros(
          (batch_size, num_frames, num_classes), dtype=np.int32
      )
      label[..., 0] = 1
      box_loss = temporal_localization_base_model.weighted_unnormalized_box_regression_loss(
          displacements, gt_displacements, label, weights, loss_type
      )
      self.assertTupleEqual(
          box_loss.shape, (batch_size, num_frames, num_classes)
      )
  
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 4 for calling the TemporalLocalizationBaseModelTest obj's underlying member method code with example usage: obj.test_weighted_unnormalized_iou_loss(weights, loss_type) without return types