Skip to content

Commit dcf674a

Browse files
committed
todo
1 parent 1b8cc3c commit dcf674a

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/onnx_ir/passes/common/clear_metadata_and_docstring_test.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,14 @@ def test_pass_with_clear_metadata_and_docstring(self):
5757
domain="my_domain",
5858
attributes=[],
5959
)
60+
func_node = ir.node(
61+
"my_function",
62+
inputs=[add_node.o(), inputs[1]],
63+
domain = "my_domain",
64+
metadata_props={"mul_key": "mul_value"},
65+
doc_string="This is a Mul node",
66+
)
67+
# TODO(justinchuby): This graph is broken. The output of the function cannot be a input to a node
6068
# Create a model with the graph and function
6169
constant_tensor = ir.tensor(np.random.rand(2, 3).astype(ir.DataType.FLOAT.numpy()))
6270
const_node = ir.node(

0 commit comments

Comments
 (0)