update test data type

apache · Dec 18, 2020 · b0d9103 · b0d9103
1 parent 67321f0
commit b0d9103
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/python-pytest/onnx/test_operators.py b/tests/python-pytest/onnx/test_operators.py
@@ -255,7 +255,7 @@ def test_onnx_export_dropout(tmp_path, dtype, p):
     x = mx.nd.array([[3,0.5,-0.5,2,7],[2,-0.4,7,3,0.2]], dtype=dtype)
     op_export_test('Dropout', M, [x], tmp_path)
 
-@pytest.mark.parametrize('dtype', ['float32', 'float64'])
+@pytest.mark.parametrize('dtype', ['float16', 'float32'])
 def test_onnx_export_softmax(tmp_path, dtype):
     x = mx.nd.random.uniform(0, 1, (2, 3, 4), dtype=dtype)
     M1 = def_model('softmax')
@@ -266,3 +266,4 @@ def test_onnx_export_softmax(tmp_path, dtype):
     M3 = def_model('softmax', use_length=True, axis=-1, temperature=0.5)
     l3 = mx.nd.array([[2,0,4],[0,0,0]], dtype=int)
     op_export_test('softmax_3', M3, [x, l3], tmp_path)
+