From 42592f05e6cfc0cd5ba443d5749a1ee560dedcd2 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Tue, 27 Aug 2019 13:31:12 -0400
Subject: [PATCH] fix
---
mhcflurry/amino_acid.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mhcflurry/amino_acid.py b/mhcflurry/amino_acid.py
index 73845bba..f8c7047b 100644
--- a/mhcflurry/amino_acid.py
+++ b/mhcflurry/amino_acid.py
@@ -153,7 +153,7 @@ def fixed_vectors_encoding(index_encoded_sequences, letter_to_vector_df):
target_shape = (
num_sequences, sequence_length, letter_to_vector_df.shape[0])
result = letter_to_vector_df.iloc[
- index_encoded_sequences.flatten()
+ index_encoded_sequences.reshape((-1,))
].values.reshape(target_shape)
return result
--
GitLab