leap-model-parser 0.1.257.dev0__tar.gz → 0.1.257.dev2__tar.gz

{leap_model_parser-0.1.257.dev0 → leap_model_parser-0.1.257.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.257.dev0
+Version: 0.1.257.dev2
 Summary: 
 Home-page: https://github.com/tensorleap/leap-model-parser
 License: MIT

{leap_model_parser-0.1.257.dev0 → leap_model_parser-0.1.257.dev2}/leap_model_parser/leap_graph_editor.py

@@ -89,6 +89,7 @@ class LeapGraphEditor:
         return None
 
     def _replace_prediction_node_name_with_correct_name(self, connections: List[NodeConnection]) -> List[NodeConnection]:
+        index_by_resolved_name: Dict[str, int] = {}
         for connection in connections:
             if connection.node_inputs is None:
                 continue
@@ -96,6 +97,15 @@ class LeapGraphEditor:
                 if 'Prediction' in input_node.type.value:
                     prediction_index = int(input_node.type.value.replace('Prediction', ''))
                     origin_name = self.keras_model.outputs[prediction_index].node.layer.name
+                    known_index = index_by_resolved_name.setdefault(origin_name, prediction_index)
+                    # The same tensor returned twice binds correctly; two tensors of one layer
+                    # (a shared layer, a multi-output op) cannot be told apart by layer name.
+                    if (self.keras_model.outputs[known_index] is not
+                            self.keras_model.outputs[prediction_index]):
+                        raise Exception(
+                            f"Prediction{known_index} and Prediction{prediction_index} both resolve to "
+                            f"layer '{origin_name}', so their bindings are ambiguous; give the model "
+                            f"outputs distinct layers")
                     input_node.name = origin_name
 
         return connections

{leap_model_parser-0.1.257.dev0 → leap_model_parser-0.1.257.dev2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "leap-model-parser"
-version = "0.1.257.dev0"
+version = "0.1.257.dev2"
 description = ""
 authors = ["idan <idan.yogev@tensorleap.ai>"]
 license = "MIT"