def hans_data_collator(features: List[InputFeatures]) -> Dict[str, torch.Tensor]:
    """
    Data collator that removes the "pairID" key if present.
    """
    batch = default_data_collator(features)
    _ = batch.pop("pairID", None)
    return batch

    Data collator that removes the "pairID" key if present.
    
dpkd/transformers/examples/research_projects/adversarial/run_hans.py