tcr_pmhc_interface_analysis.processing module

tcr_pmhc_interface_analysis.processing.annotate_tcr_pmhc_df(structure_df: DataFrame, alpha_chain_id: str | None = None, beta_chain_id: str | None = None, antigen_chain_id: str | None = None, mhc_chain1_id: str | None = None, mhc_chain2_id: str | None = None) → DataFrame

Add chain_type and cdr columns to a structure dataframe.

tcr_pmhc_interface_analysis.processing.find_anchors(cdr_df: DataFrame, structure_df: DataFrame, num_anchors: int = 1) → tuple[DataFrame, DataFrame]

Get the anchors of a cdr loop. DOES NOT SUPPORT MULTIPLE MODELS.