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 branches found
No related tags found
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