leap-model-parser 0.1.258__tar.gz → 0.1.259.dev0__tar.gz

{leap_model_parser-0.1.258 → leap_model_parser-0.1.259.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.258
+Version: 0.1.259.dev0
 Summary: 
 Home-page: https://github.com/tensorleap/leap-model-parser
 License: MIT
@@ -17,7 +17,7 @@ Requires-Dist: keras-data-format-converter (==0.1.24)
 Requires-Dist: leap-model-rebuilder (==0.1.7)
 Requires-Dist: numpy (>=1.22.3,<2.0.0)
 Requires-Dist: onnx (==1.13.0)
-Requires-Dist: onnx2kerastl (==0.0.198)
+Requires-Dist: onnx2kerastl (==0.0.199.dev0)
 Requires-Dist: tensorflow (==2.12.0) ; platform_machine == "x86_64"
 Requires-Dist: tensorflow-io-gcs-filesystem (==0.34.0)
 Requires-Dist: tensorflow-macos (==2.12.0) ; platform_machine == "arm64"

{leap_model_parser-0.1.258 → leap_model_parser-0.1.259.dev0}/leap_model_parser/leap_graph_editor.py

@@ -89,16 +89,6 @@ class LeapGraphEditor:
                 return node
         return None
 
-    @staticmethod
-    def _expected_output_count(connections: List[NodeConnection]) -> int:
-        # The integration test refers to model outputs as Prediction0, Prediction1, ...; it expects
-        # at least as many outputs as its highest index + 1.
-        indices = [int(input_node.type.value.replace('Prediction', ''))
-                   for connection in connections if connection.node_inputs is not None
-                   for input_node in connection.node_inputs.values()
-                   if 'Prediction' in input_node.type.value]
-        return max(indices) + 1 if indices else 0
-
     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:
@@ -107,12 +97,6 @@ class LeapGraphEditor:
             for input_name, input_node in connection.node_inputs.items():
                 if 'Prediction' in input_node.type.value:
                     prediction_index = int(input_node.type.value.replace('Prediction', ''))
-                    n_outputs = len(self.keras_model.outputs)
-                    if prediction_index >= n_outputs:
-                        expected = self._expected_output_count(connections)
-                        raise Exception(
-                            f"The model has {n_outputs} output{'' if n_outputs == 1 else 's'}, "
-                            f"the integration expects {expected}.")
                     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

{leap_model_parser-0.1.258 → leap_model_parser-0.1.259.dev0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "leap-model-parser"
-version = "0.1.258"
+version = "0.1.259.dev0"
 description = ""
 authors = ["idan <idan.yogev@tensorleap.ai>"]
 license = "MIT"
@@ -18,7 +18,7 @@ tensorflow = {version = "2.12.0", markers = "platform_machine  == 'x86_64'"}
 tensorflow-macos = {version = "2.12.0", markers = "platform_machine  == 'arm64'"}
 numpy = "^1.22.3"
 onnx = "1.13.0"
-onnx2kerastl = "0.0.198"
+onnx2kerastl = "0.0.199.dev0"
 keras-data-format-converter = "0.1.24"
 leap-model-rebuilder = "0.1.7"
 tensorflow-io-gcs-filesystem = "0.34.0"