leap-model-parser 0.1.187__tar.gz → 0.1.188__tar.gz

{leap_model_parser-0.1.187 → leap_model_parser-0.1.188}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.187
+Version: 0.1.188
 Summary: 
 Home-page: https://github.com/tensorleap/leap-model-parser
 License: MIT
@@ -12,7 +12,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
-Requires-Dist: code-loader (==1.0.89)
+Requires-Dist: code-loader (==1.0.91)
 Requires-Dist: keras-data-format-converter (==0.1.22)
 Requires-Dist: leap-model-rebuilder (==0.1.7)
 Requires-Dist: numpy (>=1.22.3,<2.0.0)

{leap_model_parser-0.1.187 → leap_model_parser-0.1.188}/leap_model_parser/leap_graph_editor.py

@@ -207,9 +207,9 @@ class LeapGraphEditor:
     def _handle_input_node_with_index(self, input_node: NodeMapping) -> str:
         input_index = int(input_node.type.value.replace('Input', ''))
         origin_name = self.keras_model.inputs[input_index].node.layer.name
-        input_node = self._find_input_node_by_origin_name(origin_name)
-        assert input_node is not None, f"Input node with origin name {origin_name} not found in model graph"
-        input_node_id = input_node.id
+        input_node_by_origin = self._find_input_node_by_origin_name(origin_name)
+        assert input_node_by_origin is not None, f"Input node with origin name {origin_name} not found in model graph"
+        input_node_id = input_node_by_origin.id
         self.model_graph[input_node_id].data['output_name'] = input_node.name
         output_keys = list(self.model_graph[input_node_id].outputs.keys())
         for output_key in output_keys:
@@ -246,9 +246,6 @@ class LeapGraphEditor:
 
         return input_node_id
 
-    def _find_prediction_node(self, prediction_index):
-        pass
-
     def _get_all_loss_node_ids(self):
         loss_node_ids = []
         for node_id, node_response in self.model_graph.items():

{leap_model_parser-0.1.187 → leap_model_parser-0.1.188}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "leap-model-parser"
-version = "0.1.187"
+version = "0.1.188"
 description = ""
 authors = ["idan <idan.yogev@tensorleap.ai>"]
 license = "MIT"
@@ -22,7 +22,7 @@ onnx2kerastl = "0.0.174"
 keras-data-format-converter = "0.1.22"
 leap-model-rebuilder = "0.1.7"
 tensorflow-io-gcs-filesystem = "0.34.0"
-code-loader = "1.0.89"
+code-loader = "1.0.91"
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.1.1"