From 4bb0738b73557187774b6f3a169d501d4e0490d4 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 1 Dec 2017 14:16:34 -0500
Subject: [PATCH] Fix encoding for 0-length sequences
---
mhcflurry/amino_acid.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/mhcflurry/amino_acid.py b/mhcflurry/amino_acid.py
index c9057d9a..78b3a854 100644
--- a/mhcflurry/amino_acid.py
+++ b/mhcflurry/amino_acid.py
@@ -151,11 +151,11 @@ def fixed_vectors_encoding(index_encoded_sequences, letter_to_vector_df):
-------
numpy.array of integers with shape (n, k, m)
"""
+ (num_sequences, sequence_length) = index_encoded_sequences.shape
target_shape = (
- len(index_encoded_sequences),
- len(index_encoded_sequences[0]),
- letter_to_vector_df.shape[0])
+ num_sequences, sequence_length, letter_to_vector_df.shape[0])
result = letter_to_vector_df.iloc[
index_encoded_sequences.flat
].values.reshape(target_shape)
return result
+
--
GitLab