code: def _get_states(
      labels: torch.Tensor,
      predictions: torch.Tensor,
      weights: torch.Tensor,
  ) -> Dict[str, torch.Tensor]:
      states = get_multiclass_recall_states(
          predictions,
          labels,
          weights,
          TestMulticlassRecallMetric.n_classes,
      )
      return states
  
file_path: torchrec/torchrec/metrics/tests/test_multiclass_recall.py
imports: import unittest
  from typing import Dict, Type
  import torch
  from torchrec.metrics.multiclass_recall import compute_multiclass_recall_at_k, get_multiclass_recall_states, MulticlassRecallMetric
  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 1 for calling the TestMulticlassRecallMetric obj's underlying member method code with example usage: obj._get_states(labels, predictions, weights) and returns: states