From bb2e2d54d7c27d100a944b108a90ce10140650e8 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Thu, 28 Apr 2016 23:54:46 -0400
Subject: [PATCH] AlleleData objects now have numpy arrays for all attributes
---
mhcflurry/data.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/mhcflurry/data.py b/mhcflurry/data.py
index 054c1d2e..eab00b13 100644
--- a/mhcflurry/data.py
+++ b/mhcflurry/data.py
@@ -317,9 +317,10 @@ def create_allele_data_from_peptide_to_ic50_dict(
X_binary=X_binary,
Y=Y_kmer,
ic50=ic50_array,
- peptides=kmer_peptides,
- original_peptides=original_peptides,
- original_lengths=[len(peptide) for peptide in original_peptides],
+ peptides=np.array(kmer_peptides),
+ original_peptides=np.array(original_peptides),
+ original_lengths=np.array(
+ [len(peptide) for peptide in original_peptides]),
substring_counts=counts,
weights=1.0 / counts)
--
GitLab