From dda24af51b5b44fb12b36259f4680236e8da05ed Mon Sep 17 00:00:00 2001
From: Yuanqiang Liu <liuyuanqiang.yqliu@bytedance.com>
Date: Sun, 28 Apr 2024 11:31:47 +0800
Subject: [PATCH] fix make_fx_tosa and format

---
 projects/pt1/e2e_testing/xfail_sets.py                |  1 +
 .../torch_mlir_e2e_test/test_suite/elementwise.py     | 11 ++++++-----
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/projects/pt1/e2e_testing/xfail_sets.py b/projects/pt1/e2e_testing/xfail_sets.py
index c735875c15de..87344fb99b59 100644
--- a/projects/pt1/e2e_testing/xfail_sets.py
+++ b/projects/pt1/e2e_testing/xfail_sets.py
@@ -1937,6 +1937,7 @@
     # failed to legalize operation 'torch.operator'
     "ElementwisePreluModule_basic",
     "ElementwisePreluStaticModule_basic",
+    "ElementwiseLogSigmoidModule_basic",
     # Shape Related failures
     "PrimListUnpackNumMismatchModule_basic",
     "ReshapeExpandModule_basic",
diff --git a/projects/pt1/python/torch_mlir_e2e_test/test_suite/elementwise.py b/projects/pt1/python/torch_mlir_e2e_test/test_suite/elementwise.py
index 557b524d838c..47f4a64038e9 100644
--- a/projects/pt1/python/torch_mlir_e2e_test/test_suite/elementwise.py
+++ b/projects/pt1/python/torch_mlir_e2e_test/test_suite/elementwise.py
@@ -2110,16 +2110,17 @@ def ElementwiseLogitModule_basic(module, tu: TestUtils):
 
 
 class ElementwiseLogSigmoidModule(torch.nn.Module):
-
     def __init__(self):
         super().__init__()
         self.m = torch.nn.LogSigmoid()
 
     @export
-    @annotate_args([
-        None,
-        ([-1, -1], torch.float32, True),
-    ])
+    @annotate_args(
+        [
+            None,
+            ([-1, -1], torch.float32, True),
+        ]
+    )
     def forward(self, a):
         return self.m(a)