leap-model-parser 0.1.225.dev1__tar.gz → 0.1.225.dev2__tar.gz

{leap_model_parser-0.1.225.dev1 → leap_model_parser-0.1.225.dev2}/PKG-INFO

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

{leap_model_parser-0.1.225.dev1 → leap_model_parser-0.1.225.dev2}/leap_model_parser/contract/graph.py

@@ -47,7 +47,7 @@ class Node:
     outputs: Dict[str, ConnectionOutput] = field(default_factory=dict)
     pruning_plan_id: Optional[str] = None
     wrapper: Optional[WrapperData] = None
-    shape: Optional[List[Optional[int]]] = None
+    shape: Optional[List[Optional[str]]] = None
 
     def __key(self):
         return (self.id, self.name)

{leap_model_parser-0.1.225.dev1 → leap_model_parser-0.1.225.dev2}/leap_model_parser/keras_json_model_import.py

@@ -220,11 +220,17 @@ class KerasJsonModelImport:
 
         self.layer_data_adjustments(data, layer_metadata)
         node = Node(id=str(self.id), name=layer_metadata.get(
-            "class_name", layer["class_name"]), data=data, position=[0, 0], shape=layer['output_shape'])
+            "class_name", layer["class_name"]), data=data, position=[0, 0],
+                    shape=self._convert_layer_shape_to_string(layer['output_shape']))
         if 'wrapper' in layer:
             node.wrapper = layer['wrapper']
         self.nodes_cache[node_key] = node
 
+    @staticmethod
+    def _convert_layer_shape_to_string(s: List[Optional[int]]) -> List[str]:
+        return [str(dim) if dim is not None else 'dynamic_shape' for dim in s]
+
+
     @classmethod
     def handle_wrapper_layer(cls, layer):
         wrapped_layer = layer['config']['layer']

{leap_model_parser-0.1.225.dev1 → leap_model_parser-0.1.225.dev2}/pyproject.toml

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