add a comment about shape test

apache · Dec 27, 2018 · ed8db55 · ed8db55
1 parent e958003
commit ed8db55
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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)