fixup! Add Converter.layers attribute

nengo · Feb 28, 2020 · ca957e5 · ca957e5
1 parent b8647c7
commit ca957e5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nengo_dl/converter.py b/nengo_dl/converter.py
@@ -362,9 +362,9 @@ def __init__(self):
 
         def _get_key(self, key):
             if isinstance(key, tf.keras.layers.Layer):
-                if (
-                    len(key.inbound_nodes) > 1
-                    or len(key.inbound_nodes[0].output_tensors) > 1
+                if len(key.inbound_nodes) > 1 or (
+                    isinstance(key.inbound_nodes[0].output_tensors, list)
+                    and len(key.inbound_nodes[0].output_tensors) > 1
                 ):
                     raise KeyError(
                         "Layer %s is ambiguous because it has multiple output tensors; "