leap-model-parser 0.1.250.dev7__tar.gz → 0.1.250.dev8__tar.gz

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 (26) hide show
  1. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/PKG-INFO +1 -1
  2. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/torch_utils.py +3 -0
  3. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/pyproject.toml +1 -1
  4. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/LICENSE +0 -0
  5. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/README.md +0 -0
  6. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/__init__.py +0 -0
  7. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/contract/__init__.py +0 -0
  8. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/contract/graph.py +0 -0
  9. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/contract/importmodelresponse.py +0 -0
  10. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/contract/nodedata.py +0 -0
  11. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/contract/ui_components.json +0 -0
  12. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/keras_json_model_import.py +0 -0
  13. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/leap_graph_editor.py +0 -0
  14. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/model_parser.py +0 -0
  15. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/torch_graph_editor.py +0 -0
  16. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/torch_model_parser.py +0 -0
  17. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/utils/__init__.py +0 -0
  18. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/utils/layerpedia/__init__.py +0 -0
  19. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/utils/layerpedia/layerpedia.py +0 -0
  20. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/utils/tlinspection/__init__.py +0 -0
  21. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/utils/tlinspection/leapinspection.py +0 -0
  22. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/utils/uicomponents/__init__.py +0 -0
  23. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/utils/uicomponents/generatenodedata.py +0 -0
  24. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/utils/uicomponents/tensorflowinscpection.py +0 -0
  25. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/utils/uicomponents/ui_components.json +0 -0
  26. {leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/utils/uicomponents/ui_components_config.yaml +0 -0
{leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: leap-model-parser
3
- Version: 0.1.250.dev7
3
+ Version: 0.1.250.dev8
4
4
  Summary:
5
5
  Home-page: https://github.com/tensorleap/leap-model-parser
6
6
  License: MIT
{leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/leap_model_parser/torch_utils.py RENAMED
@@ -26,10 +26,13 @@ def get_full_args(ep: ExportedProgram, user_inputs: List[torch.Tensor]) -> List[
26
26
 
27
27
 
28
28
  def build_module_to_node_map(ep: ExportedProgram) -> Dict[str, Tuple[str, str]]:
29
+ import operator as _op
29
30
  mapping: Dict[str, Tuple[str, str]] = {}
30
31
  for node in ep.graph_module.graph.nodes:
31
32
  if node.op != "call_function":
32
33
  continue
34
+ if node.target is _op.getitem:
35
+ continue
33
36
  stack = node.meta.get("nn_module_stack", {})
34
37
  if not stack:
35
38
  continue
{leap_model_parser-0.1.250.dev7 → leap_model_parser-0.1.250.dev8}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "leap-model-parser"
3
- version = "0.1.250.dev7"
3
+ version = "0.1.250.dev8"
4
4
  description = ""
5
5
  authors = ["idan <idan.yogev@tensorleap.ai>"]
6
6
  license = "MIT"