Skip to content

Add BigMHC paired prediction API - #257

Merged
iskandr merged 1 commit into
masterfrom
agent/bigmhc-predict-pairs
Jul 10, 2026
Merged

Add BigMHC paired prediction API#257
iskandr merged 1 commit into
masterfrom
agent/bigmhc-predict-pairs

Conversation

@iskandr

@iskandr iskandr commented Jul 10, 2026

Copy link
Copy Markdown
Contributor

Summary

Fixes #254.

Adds a row-wise BigMHC.predict_pairs(...) API for explicit peptide/allele pairs. Callers can pass parallel peptides, alleles lists or an iterable of (peptide, allele) rows; results are returned in input order as one PeptideResult per pair.

Impact

Paired neoantigen scoring no longer needs one BigMHC instance per allele or an O(N x A) cross product with diagonal filtering. The existing predict(peptides) cross-product API now reuses the paired path and keeps its previous return shape.

Also adds predict_pairs_dataframe(...) for the row-wise API.

Validation

  • pytest tests/test_bigmhc_pairs.py -q
  • ./lint.sh
  • ./test.sh (642 passed, 51 skipped, 2 xfailed)

Version

Bumped package version from 3.31.1 to 3.31.2.

@iskandr
iskandr marked this pull request as ready for review July 10, 2026 23:38
@iskandr
iskandr merged commit 272bcf2 into master Jul 10, 2026
5 checks passed
@iskandr
iskandr deleted the agent/bigmhc-predict-pairs branch July 10, 2026 23:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

BigMHC: add row-wise / paired (peptide, allele) prediction; public API only does peptide × allele cross product

1 participant