From e9cc606aa1bdd8c69c0deb60a09a93ba947765fc Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 13 Sep 2019 13:44:02 -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 eb621f8c..411ba54d 100644
--- a/mhcflurry/class1_neural_network.py
+++ b/mhcflurry/class1_neural_network.py
@@ -1321,7 +1321,7 @@ class Class1NeuralNetwork(object):
peptide_allele_merge_activation)(current_layer)
densenet_layers = [] if topology == "densenet" else None
- for (i, layer_size) in enumerate(layer_size):
+ for (i, layer_size) in enumerate(layer_sizes):
if densenet_layers is not None:
densenet_layers.append(current_layer)
if len(densenet_layers) > 1:
--
GitLab