From 6dab15f6443bd8c5518c3bf758f655002295bf31 Mon Sep 17 00:00:00 2001 From: Vandana Kannan Date: Thu, 27 Dec 2018 09:33:27 -0800 Subject: [PATCH] add a comment about shape test --- tests/python-pytest/onnx/test_node.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/python-pytest/onnx/test_node.py b/tests/python-pytest/onnx/test_node.py index 81716bf6a924..6a0f8bcd73c2 100644 --- a/tests/python-pytest/onnx/test_node.py +++ b/tests/python-pytest/onnx/test_node.py @@ -204,6 +204,8 @@ def get_onnx_graph(testname, input_names, inputs, output_name, output_shape, att {'kernel': (4, 5), 'pad': (0, 0), 'stride': (1, 1), 'p_value': 2, 'pool_type': 'lp', 'global_pool': True}, False, {'modify': {'p_value': 'p'}, 'remove': ['pool_type', 'kernel', 'pad', 'stride', 'global_pool']}, True, False), + + # since results would be random, checking for shape alone ("test_multinomial", mx.sym.sample_multinomial, "Multinomial", [np.array([0, 0.1, 0.2, 0.3, 0.4]).astype("float32")], {'shape': (10,)}, False, {'modify': {'shape': 'sample_size'}}, False, True)