From 3afc2bfa73b5a217403a89ab9403b93c01e25fbe Mon Sep 17 00:00:00 2001 From: Oliver Rausch Date: Wed, 16 Sep 2020 09:25:51 +0200 Subject: [PATCH] Remove mutable arguments from symbolic_shape_infer (#5166) --- .../core/providers/nuphar/scripts/symbolic_shape_infer.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/onnxruntime/core/providers/nuphar/scripts/symbolic_shape_infer.py b/onnxruntime/core/providers/nuphar/scripts/symbolic_shape_infer.py index 02062180b7..41a641b867 100755 --- a/onnxruntime/core/providers/nuphar/scripts/symbolic_shape_infer.py +++ b/onnxruntime/core/providers/nuphar/scripts/symbolic_shape_infer.py @@ -34,7 +34,8 @@ def handle_negative_axis(axis, rank): assert axis < rank and axis >= -rank return axis if axis >= 0 else rank + axis -def get_opset(mp, domain=['', 'onnx', 'ai.onnx']): +def get_opset(mp, domain=None): + domain = domain or ['', 'onnx', 'ai.onnx'] if type(domain) != list: domain = [domain] for opset in mp.opset_import: @@ -1128,8 +1129,8 @@ class SymbolicShapeInference: vi = self.known_vi_[node.output[0]] vi.CopyFrom(new_vi) - def _infer_impl(self, in_mp, start_sympy_data={}): - self.sympy_data_ = start_sympy_data + def _infer_impl(self, in_mp, start_sympy_data=None): + self.sympy_data_ = start_sympy_data or {} self.out_mp_.graph.ClearField('value_info') self._apply_suggested_merge(graph_input_only=True) self.input_symbols_ = set()