code: def _compute(states: Dict[str, torch.Tensor]) -> torch.Tensor:
      return get_mean(
          states["weighted_sum"], states["weighted_num_samples"]
      )
  
file_path: torchrec/torchrec/metrics/tests/test_weighted_avg.py
imports: import unittest
  from typing import Dict, Iterable, Optional, Type, Union
  import torch
  from torchrec.metrics.rec_metric import RecComputeMode, RecMetric, RecTaskInfo
  from torchrec.metrics.test_utils import metric_test_helper, rec_metric_value_test_launcher, TestMetric
  from torchrec.metrics.weighted_avg import get_mean, WeightedAvgMetric
label_desc: Use image node_id 2 for calling the TestWeightedAvgMetric obj's underlying member method code with example usage: obj._compute(states) and returns: get_mean