Skip to content

Commit 9d73e16

Browse files
committed
[Test] Remove "allowed values" check for data layouts CustomOp
1 parent 817a23e commit 9d73e16

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/core/test_custom_onnx_exec.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,9 @@ def test_execute_custom_node_multithreshold():
274274
assert (execution_context["out"] == outputs_nhwc).all()
275275
# check the set of allowed values
276276
op_inst = getCustomOp(node_def)
277-
assert op_inst.get_nodeattr_allowed_values("data_layout") == {"NCHW", "NHWC", "NC", "NWC", "NCW"}
277+
# TODO: Removed this check to generalize the supported data layouts, but do
278+
# we need some other check to verify the validity of data layouts?
279+
# assert op_inst.get_nodeattr_allowed_values("data_layout") == {"NCHW", "NHWC", "NC", "NWC", "NCW"}
278280
# exercise the allowed value checks
279281
# try to set attribute to non-allowed value, should raise an exception
280282
try:

0 commit comments

Comments
 (0)