def METHOD_NAME(
        self,
        feats: Tuple[torch.Tensor],
        data_samples: List[ReIDDataSample] = None) -> List[ReIDDataSample]:
    """Inference without augmentation.
    Args:
        feats (Tuple[Tensor]): The features extracted from the backbone.
            Multiple stage inputs are acceptable but only the last stage
            will be used.
        data_samples (List[ReIDDataSample], optional): The annotation
            data of every samples. If not None, set ``pred_label`` of
            the input data samples. Defaults to None.
    Returns:
        List[ReIDDataSample]: A list of data samples which contains the
        predicted results.
    """
    # The part can be traced by torch.fx
    feats = self(feats)
    # The part can not be traced by torch.fx
    data_samples = self.predict_by_feat(feats, data_samples)
    return data_samples