Fix a lint issue and correct xfail.

microsoft · Dec 1, 2023 · 3d11fbf · 3d11fbf
1 parent f52ff11
commit 3d11fbf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git 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
@@ -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",
     ),