diff --git a/onnxscript/tests/functions/onnxfns2_test.py b/onnxscript/tests/functions/onnxfns2_test.py index 9bcbf69f4..f9dc5c9f0 100644 --- a/onnxscript/tests/functions/onnxfns2_test.py +++ b/onnxscript/tests/functions/onnxfns2_test.py @@ -16,10 +16,8 @@ def test_onnxfns_reduce_sum_square(self): self.run_onnx_test( onnxfns2.ReduceSumSquare, keepdims=default_keepdims, - # default attributes are not supported yet. skip_test_names=[ - "test_reduce_sum_square_default_axes_keepdims_example", - "test_reduce_sum_square_default_axes_keepdims_random", + "test_reduce_sum_square_empty_set", ], ) @@ -29,10 +27,8 @@ def test_onnxfns_reduce_l1(self): self.run_onnx_test( onnxfns2.ReduceL1, keepdims=default_keepdims, - # default attributes are not supported yet. skip_test_names=[ - "test_reduce_l1_default_axes_keepdims_example", - "test_reduce_l1_default_axes_keepdims_random", + "test_reduce_l1_empty_set", ], ) @@ -42,10 +38,8 @@ def test_onnxfns_reduce_l2(self): self.run_onnx_test( onnxfns2.ReduceL2, keepdims=default_keepdims, - # default attributes are not supported yet. skip_test_names=[ - "test_reduce_l2_default_axes_keepdims_example", - "test_reduce_l2_default_axes_keepdims_random", + "test_reduce_l2_empty_set", ], ) @@ -55,8 +49,9 @@ def test_onnxfns_reduce_log_sum(self): self.run_onnx_test( onnxfns2.ReduceLogSum, keepdims=default_keepdims, - # default attributes are not supported yet. - skip_test_names=["test_reduce_log_sum_default"], + skip_test_names=[ + "test_reduce_log_sum_empty_set", + ], ) def test_onnxfns_reduce_log_sum_exp(self): @@ -65,10 +60,8 @@ def test_onnxfns_reduce_log_sum_exp(self): self.run_onnx_test( onnxfns2.ReduceLogSumExp, keepdims=default_keepdims, - # default attributes are not supported yet. skip_test_names=[ - "test_reduce_log_sum_exp_default_axes_keepdims_example", - "test_reduce_log_sum_exp_default_axes_keepdims_random", + "test_reduce_log_sum_exp_empty_set", ], ) diff --git a/requirements/ci/requirements-onnx-weekly.txt b/requirements/ci/requirements-onnx-weekly.txt index c217b399a..dc98af02f 100644 --- a/requirements/ci/requirements-onnx-weekly.txt +++ b/requirements/ci/requirements-onnx-weekly.txt @@ -1 +1 @@ -onnx-weekly==1.15.0.dev20230807 +onnx-weekly==1.15.0.dev20231002