leap-model-parser 0.1.225.dev1__py3-none-any.whl → 0.1.225.dev2__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. leap_model_parser/contract/graph.py +1 -1
  2. leap_model_parser/keras_json_model_import.py +7 -1
  3. {leap_model_parser-0.1.225.dev1.dist-info → leap_model_parser-0.1.225.dev2.dist-info}/METADATA +1 -1
  4. {leap_model_parser-0.1.225.dev1.dist-info → leap_model_parser-0.1.225.dev2.dist-info}/RECORD +7 -7
  5. {leap_model_parser-0.1.225.dev1.dist-info → leap_model_parser-0.1.225.dev2.dist-info}/LICENSE +0 -0
  6. {leap_model_parser-0.1.225.dev1.dist-info → leap_model_parser-0.1.225.dev2.dist-info}/WHEEL +0 -0
  7. {leap_model_parser-0.1.225.dev1.dist-info → leap_model_parser-0.1.225.dev2.dist-info}/entry_points.txt +0 -0
leap_model_parser/contract/graph.py CHANGED
@@ -47,7 +47,7 @@ class Node:
47
47
  outputs: Dict[str, ConnectionOutput] = field(default_factory=dict)
48
48
  pruning_plan_id: Optional[str] = None
49
49
  wrapper: Optional[WrapperData] = None
50
- shape: Optional[List[Optional[int]]] = None
50
+ shape: Optional[List[Optional[str]]] = None
51
51
 
52
52
  def __key(self):
53
53
  return (self.id, self.name)
leap_model_parser/keras_json_model_import.py CHANGED
@@ -220,11 +220,17 @@ class KerasJsonModelImport:
220
220
 
221
221
  self.layer_data_adjustments(data, layer_metadata)
222
222
  node = Node(id=str(self.id), name=layer_metadata.get(
223
- "class_name", layer["class_name"]), data=data, position=[0, 0], shape=layer['output_shape'])
223
+ "class_name", layer["class_name"]), data=data, position=[0, 0],
224
+ shape=self._convert_layer_shape_to_string(layer['output_shape']))
224
225
  if 'wrapper' in layer:
225
226
  node.wrapper = layer['wrapper']
226
227
  self.nodes_cache[node_key] = node
227
228
 
229
+ @staticmethod
230
+ def _convert_layer_shape_to_string(s: List[Optional[int]]) -> List[str]:
231
+ return [str(dim) if dim is not None else 'dynamic_shape' for dim in s]
232
+
233
+
228
234
  @classmethod
229
235
  def handle_wrapper_layer(cls, layer):
230
236
  wrapped_layer = layer['config']['layer']
{leap_model_parser-0.1.225.dev1.dist-info → leap_model_parser-0.1.225.dev2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: leap-model-parser
3
- Version: 0.1.225.dev1
3
+ Version: 0.1.225.dev2
4
4
  Summary:
5
5
  Home-page: https://github.com/tensorleap/leap-model-parser
6
6
  License: MIT
{leap_model_parser-0.1.225.dev1.dist-info → leap_model_parser-0.1.225.dev2.dist-info}/RECORD RENAMED
@@ -1,11 +1,11 @@
1
1
  LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
2
2
  leap_model_parser/__init__.py,sha256=OAU7rFHAVVWUM-cDtQ4Ohum567KN8M-YTkHZp5KiYbo,132
3
3
  leap_model_parser/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
- leap_model_parser/contract/graph.py,sha256=uE0WecEB8WjDFNMZbRLEzmYq7w8nnyjiLU5iA9O9eqw,1180
4
+ leap_model_parser/contract/graph.py,sha256=3cTq2eWM_b5VNyOQnwtBGgNW4004jzgkfsVeKAkLmmg,1180
5
5
  leap_model_parser/contract/importmodelresponse.py,sha256=GlvnKS8xrebU2Sj0dxqtEhAOOo3DlOtT7AMJ2BlzH9E,145
6
6
  leap_model_parser/contract/nodedata.py,sha256=1_ML0nzp3QUZ0_9mGSLhfO4_hqjYMwi0DWLwymUnWEs,43326
7
7
  leap_model_parser/contract/ui_components.json,sha256=0lsxwOLElW1E-imCcdh3zKPWgzFuQ_bApG6aHvYfTvo,410591
8
- leap_model_parser/keras_json_model_import.py,sha256=Bt1wZX5PK6YFUKGk2wNq03M5sckeBb3HdDpzMpMGWrA,16819
8
+ leap_model_parser/keras_json_model_import.py,sha256=cxDaJdztGnsSblcQBCN0OAl2Zg6bCqrIimsJ4DcNZeE,17052
9
9
  leap_model_parser/leap_graph_editor.py,sha256=0SDHNVkMq94PXWaTqZzMlzJMRLWebOj4uyz64c_KCTQ,13559
10
10
  leap_model_parser/model_parser.py,sha256=ovXrVQqdbWJPMikJp96Llim-8fmcVM_Udoif5Tcf_kg,6949
11
11
  leap_model_parser/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -18,8 +18,8 @@ leap_model_parser/utils/uicomponents/generatenodedata.py,sha256=LRaPlO5jJ9pUtkvL
18
18
  leap_model_parser/utils/uicomponents/tensorflowinscpection.py,sha256=ym613z9iQKPDBpr0RYD35bTABdm1L-Ez86G47BYT7qw,6775
19
19
  leap_model_parser/utils/uicomponents/ui_components.json,sha256=0lsxwOLElW1E-imCcdh3zKPWgzFuQ_bApG6aHvYfTvo,410591
20
20
  leap_model_parser/utils/uicomponents/ui_components_config.yaml,sha256=cRH8T-c3TAL0nfefRvt9pFsjbTWNEg38NRyHR7RpJsk,19534
21
- leap_model_parser-0.1.225.dev1.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
22
- leap_model_parser-0.1.225.dev1.dist-info/METADATA,sha256=RK88uZWeYk9yba7RknolL0wNTsI72PK0pIP8_Ur9e5g,1109
23
- leap_model_parser-0.1.225.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
24
- leap_model_parser-0.1.225.dev1.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
25
- leap_model_parser-0.1.225.dev1.dist-info/RECORD,,
21
+ leap_model_parser-0.1.225.dev2.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
22
+ leap_model_parser-0.1.225.dev2.dist-info/METADATA,sha256=OtidGaxRHm4tC9PWOGTWmA3RudmQxMeWe7fWKYZCdLc,1109
23
+ leap_model_parser-0.1.225.dev2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
24
+ leap_model_parser-0.1.225.dev2.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
25
+ leap_model_parser-0.1.225.dev2.dist-info/RECORD,,