summaryrefslogtreecommitdiff
path: root/nn/runtime/test/specs/V1_3/if_constant.mod.py
diff options
context:
space:
mode:
Diffstat (limited to 'nn/runtime/test/specs/V1_3/if_constant.mod.py')
-rw-r--r--nn/runtime/test/specs/V1_3/if_constant.mod.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/nn/runtime/test/specs/V1_3/if_constant.mod.py b/nn/runtime/test/specs/V1_3/if_constant.mod.py
index badf4476b..ae6750146 100644
--- a/nn/runtime/test/specs/V1_3/if_constant.mod.py
+++ b/nn/runtime/test/specs/V1_3/if_constant.mod.py
@@ -24,8 +24,8 @@ output_data = {
False: [x - y for (x, y) in zip(x_data, y_data)],
}
-ValueType = ["TENSOR_FLOAT32", "{3, 4}"]
-BoolType = ["TENSOR_BOOL8", "{1}"]
+ValueType = ["TENSOR_FLOAT32", [3, 4]]
+BoolType = ["TENSOR_BOOL8", [1]]
def MakeBranchModel(operation_name):
x = Input("x", ValueType)