From 3d11fbfb443cda28bb307d22163f1a50fc5c247d Mon Sep 17 00:00:00 2001 From: Jay Zhang Date: Fri, 1 Dec 2023 19:59:29 +0800 Subject: [PATCH] Fix a lint issue and correct xfail. --- onnxscript/tests/function_libs/torch_lib/ops_test.py | 4 ++-- onnxscript/tests/function_libs/torch_lib/ops_test_data.py | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/onnxscript/tests/function_libs/torch_lib/ops_test.py b/onnxscript/tests/function_libs/torch_lib/ops_test.py index 47957e1b4..0913267be 100644 --- a/onnxscript/tests/function_libs/torch_lib/ops_test.py +++ b/onnxscript/tests/function_libs/torch_lib/ops_test.py @@ -318,7 +318,7 @@ def test_output_match_opinfo_( @ops_test_common.add_decorate_info( ops_test_data.OPS_DB, "TestOutputConsistencyEager", - "test_output_match_opinfo_", + "test_complex_output_match_opinfo_", skip_or_xfails=ops_test_data.EXPECTED_SKIPS_OR_FAILS, ) @common_device_type.ops( # type: ignore[misc] @@ -380,7 +380,7 @@ def test_output_match_opinfo_( @ops_test_common.add_decorate_info( ops_test_data.OPS_DB, "TestOutputConsistencyFullGraph", - "test_output_match_opinfo_", + "test_complex_output_match_opinfo_", skip_or_xfails=ops_test_data.EXPECTED_SKIPS_OR_FAILS, ) @common_device_type.ops( # type: ignore[misc] diff --git a/onnxscript/tests/function_libs/torch_lib/ops_test_data.py b/onnxscript/tests/function_libs/torch_lib/ops_test_data.py index dcf75bb95..8f6d370fc 100644 --- a/onnxscript/tests/function_libs/torch_lib/ops_test_data.py +++ b/onnxscript/tests/function_libs/torch_lib/ops_test_data.py @@ -472,8 +472,7 @@ def _where_input_wrangler( tolerance={torch.complex64: (3e-3, 1.8e-4)}, trace_only=True, complex=True, - ) - .xfail( + ).xfail( dtypes=(torch.complex64,), reason="fixme: the result is wrong: https://github.com/microsoft/onnxscript/pull/926", ),