Remove msg from assert_close

Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

Author: larryliu0820

extension/llm/modules/test/test_attention.py

@@ -95,11 +95,7 @@ def test_attention_eager(self):
         et_res = self.et_mha(self.x, self.x)  # Self attention.
         tt_res = self.tt_mha(self.x, self.x)  # Self attention.
 
-        assert_close(
-            et_res,
-            tt_res,
-            msg=f"TorchTune output is not close to ET output.\n\nTorchTune: {tt_res}\nET output: {et_res}",
-        )
+        assert_close(et_res, tt_res)
 
         # test with kv cache
         self.et_mha.setup_cache(1, dtype=torch.float32, max_seq_len=20)
@@ -130,11 +126,7 @@ def test_attention_eager(self):
             self.x, self.x, input_pos=next_input_pos
         )  # Self attention with input pos.
 
-        assert_close(
-            et_res,
-            tt_res,
-            msg=f"TorchTune output is not close to ET output.\n\nTorchTune: {tt_res}\nET output: {et_res}",
-        )
+        assert_close(et_res, tt_res)
 
     def test_attention_export(self):
         # Self attention.
@@ -147,11 +139,7 @@ def test_attention_export(self):
         et_res = et_mha_ep.module()(self.x, self.x, input_pos=self.input_pos)
         tt_res = self.tt_mha(self.x, self.x, input_pos=self.input_pos)
 
-        assert_close(
-            et_res,
-            tt_res,
-            msg=f"TorchTune output is not close to ET output.\n\nTorchTune: {tt_res}\nET output: {et_res}",
-        )
+        assert_close(et_res, tt_res)
 
         # TODO: KV cache.
 
@@ -177,10 +165,6 @@ def test_attention_executorch(self):
         et_res = method.execute((self.x, self.x, self.input_pos))
         tt_res = self.tt_mha(self.x, self.x, input_pos=self.input_pos)
 
-        assert_close(
-            et_res[0],
-            tt_res,
-            msg=f"TorchTune output is not close to ET output.\n\nTorchTune: {tt_res}\nET output: {et_res[0]}",
-        )
+        assert_close(et_res[0], tt_res)
 
         # TODO: KV cache.