Skip to content
Snippets Groups Projects
Commit f0cc4e48 authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

fix

parent 306237ce
No related merge requests found
......@@ -1120,15 +1120,14 @@ class Class1AffinityPredictor(object):
masked_allele_encoding = AlleleEncoding(
df.normalized_allele,
borrow_from=master_allele_encoding)
masked_peptides = peptides
masked_peptides = EncodableSequences.create(peptides)
elif mask.sum() > 0:
row_slice = mask
masked_allele_encoding = AlleleEncoding(
df.loc[mask].normalized_allele,
borrow_from=master_allele_encoding)
masked_peptides = peptides.sequences[mask]
masked_peptides = EncodableSequences.create(masked_peptides)
masked_peptides = EncodableSequences.create(
peptides.sequences[mask])
if row_slice is not None:
# The following line is a performance optimization that may be
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment