code: def test_moment_retrieval_model_loss_function(
      self, cls_loss_type, box_loss_type, all_gather_loss
  ):
      config = ml_collections.ConfigDict(
          {
              "classification_loss_type": cls_loss_type,
              "box_loss_type": box_loss_type,
              "all_gather_loss": all_gather_loss,
          }
      )
      model = MockMomentRetrievalModel(config)
      loss = model.loss_function(self.logits, self.batch)
      self.assertGreater(loss, 0.0)
  
file_path: scenic/scenic/projects/unloc/moment_retrieval_base_model_test.py
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 moment_retrieval_base_model
label_desc: Use image node_id 2 for calling the MomentRetrievalBaseModelTest obj's underlying member method code with example usage: obj.test_moment_retrieval_model_loss_function(cls_loss_type, box_loss_type, all_gather_loss) without return types