code: def _compute(states: Dict[str, torch.Tensor]) -> torch.Tensor:
      return compute_mse(
          states["error_sum"],
          states["weighted_num_samples"],
      )
  
file_path: torchrec/torchrec/metrics/tests/test_mse.py
imports: import unittest
  from typing import Dict, Type
  import torch
  from torchrec.metrics.mse import compute_mse, compute_rmse, MSEMetric
  from torchrec.metrics.rec_metric import RecComputeMode, RecMetric
  from torchrec.metrics.test_utils import metric_test_helper, rec_metric_value_test_launcher, TestMetric
label_desc: Use image node_id 2 for calling the TestMSEMetric obj's underlying member method code with example usage: obj._compute(states) and returns: compute_mse