From 70ce58298cdc5e6d480c292e3fc5822e140ea483 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Tue, 19 Jun 2018 12:37:58 -0400
Subject: [PATCH] fix
---
mhcflurry/class1_neural_network.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py
index 582dac78..6a8bb96b 100644
--- a/mhcflurry/class1_neural_network.py
+++ b/mhcflurry/class1_neural_network.py
@@ -884,7 +884,7 @@ class Class1NeuralNetwork(object):
input_dim=allele_representations.shape[0],
output_dim=allele_representations.shape[1],
input_length=1,
- trainable=False)
+ trainable=False)(allele_input)
allele_layer = Flatten(name="allele_flat")(allele_representation)
--
GitLab