Skip to content

GitLab

Unverified Commit 10c42663 authored by Chris Austen's avatar Chris Austen Committed by GitHub
Browse files

replace np.bool with bool as per numpy request (#1640)

parent 49b341d3
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
test/onnx/gen_onnx.py
View file @ 10c42663
...@@ -2612,7 +2612,7 @@ def if_else_test_inlined(): ...@@ -2612,7 +2612,7 @@ def if_else_test_inlined():
else_body = onnx.helper.make_graph([else_mul_node], 'else_body', [], else_body = onnx.helper.make_graph([else_mul_node], 'else_body', [],
[else_out]) [else_out])
cond = np.array([0]).astype(np.bool) cond = np.array([0]).astype(bool)
cond_tensor = helper.make_tensor(name="cond", cond_tensor = helper.make_tensor(name="cond",
data_type=TensorProto.BOOL, data_type=TensorProto.BOOL,
dims=cond.shape, dims=cond.shape,
...@@ -2682,7 +2682,7 @@ def if_then_else_multi_output_shapes_inlined_test(): ...@@ -2682,7 +2682,7 @@ def if_then_else_multi_output_shapes_inlined_test():
else_body = onnx.helper.make_graph([else_mul_node, else_sub_node], else_body = onnx.helper.make_graph([else_mul_node, else_sub_node],
'else_body', [], [else_out, else_out2]) 'else_body', [], [else_out, else_out2])
cond = np.array([1]).astype(np.bool) cond = np.array([1]).astype(bool)
cond_tensor = helper.make_tensor(name="cond", cond_tensor = helper.make_tensor(name="cond",
data_type=TensorProto.BOOL, data_type=TensorProto.BOOL,
dims=cond.shape, dims=cond.shape,
...@@ -3117,7 +3117,7 @@ def if_then_test_inlined(): ...@@ -3117,7 +3117,7 @@ def if_then_test_inlined():
else_body = onnx.helper.make_graph([else_mul_node], 'else_body', [], else_body = onnx.helper.make_graph([else_mul_node], 'else_body', [],
[else_out]) [else_out])
cond = np.array([1]).astype(np.bool) cond = np.array([1]).astype(bool)
cond_tensor = helper.make_tensor(name="cond", cond_tensor = helper.make_tensor(name="cond",
data_type=TensorProto.BOOL, data_type=TensorProto.BOOL,
dims=cond.shape, dims=cond.shape,
...@@ -6902,7 +6902,7 @@ def sum_type_test(): ...@@ -6902,7 +6902,7 @@ def sum_type_test():
t_bool = helper.make_tensor(name="bool", t_bool = helper.make_tensor(name="bool",
data_type=TensorProto.BOOL, data_type=TensorProto.BOOL,
dims=valb.shape, dims=valb.shape,
vals=valb.astype(np.bool)) vals=valb.astype(bool))
val = np.array([1, 1]) val = np.array([1, 1])
t_int8 = helper.make_tensor(name="int8", t_int8 = helper.make_tensor(name="int8",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment