Skip to content

fix: export: added additional arguments on squeeze and unsqueeze

Charles Villard requested to merge silvanosky/aidge_onnx:fix_export_squeeze into dev

Context

When exporting onnx networks error :

TypeError: export_unsqueeze() got an unexpected keyword argument 'enable_custom_op'

Modified files

  • aidge_onnx/node_export/aidge_converters/squeeze.py
  • aidge_onnx/node_export/aidge_converters/unsqueeze.py

Detailed major modifications

Added variadic parameter to handle when additionnal unused parameters are provided on squeeze and unsqueeze

Merge request reports

Copyright © Eclipse Foundation, Inc. All Rights Reserved.     Privacy Policy | Terms of Use | Copyright Agent