def update()

in core/metric_mixin.py [0:0]

• 6 lines of code
• 5 McCabe index (conditional complexity)

  def update(self, outputs: Dict[str, torch.Tensor]):
    results = self.transform(outputs)
    # Do not try to update if any tensor is empty as a result of stratification.
    for value in results.values():
      if torch.is_tensor(value) and not value.nelement():
        return
    super().update(**results)