From 2c25d99a44e41af6049e2ca305cdefa6c18d9575 Mon Sep 17 00:00:00 2001
From: Alex Rubinsteyn <alex.rubinsteyn@gmail.com>
Date: Wed, 1 Jul 2015 18:08:23 -0400
Subject: [PATCH] added missing class1 file

---
 mhcflurry/class1.py | 11 +++++++++++
 1 file changed, 11 insertions(+)
 create mode 100644 mhcflurry/class1.py

diff --git a/mhcflurry/class1.py b/mhcflurry/class1.py
new file mode 100644
index 00000000..f90e79ec
--- /dev/null
+++ b/mhcflurry/class1.py
@@ -0,0 +1,11 @@
+import pandas as pd
+
+from .mhc1_binding_predictor import Mhc1BindingPredictor
+
+def predict(alleles, peptides):
+    allele_dataframes = []
+    for allele in alleles:
+        model = Mhc1BindingPredictor(allele=allele)
+        df = model.predict_peptides(peptides)
+        allele_dataframes.append(df)
+    return pd.concat(allele_dataframes)
-- 
GitLab