From b3319ab4fbffe68f9c2b0c2de8c85b9d296f0773 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Tue, 19 Jun 2018 12:41:15 -0400
Subject: [PATCH] fix
---
mhcflurry/class1_neural_network.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py
index b6c4945d..b185fdba 100644
--- a/mhcflurry/class1_neural_network.py
+++ b/mhcflurry/class1_neural_network.py
@@ -955,4 +955,6 @@ class Class1NeuralNetwork(object):
if existing.shape == allele_representations.shape:
layer.set_weights([allele_representations])
else:
- raise NotImplementedError("Network surgery required")
+ raise NotImplementedError(
+ "Network surgery required: %s != %s" % (
+ str(existing), str(allele_representations.shape)))
--
GitLab