diff options
Diffstat (limited to 'ethosu/vela/test/testutil.py')
| -rw-r--r-- | ethosu/vela/test/testutil.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ethosu/vela/test/testutil.py b/ethosu/vela/test/testutil.py index 4b2938b9..9ba39bc5 100644 --- a/ethosu/vela/test/testutil.py +++ b/ethosu/vela/test/testutil.py @@ -107,11 +107,12 @@ def create_op_with_quant_tensors( return op -def create_op(op_type, inputs, output, attrs=dict()): +def create_op(op_type, inputs, output, attrs=None): op = Operation(op_type, output.name + "_op") op.inputs = inputs op.outputs = [output] - op.attrs = attrs + if attrs is not None: + op.attrs = attrs return op |