From 52a57f7c7cfdc9337ac9dbcf4be94f25fd61784e Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 16 Mar 2023 12:48:52 +0100 Subject: [PATCH] Update expected values in `MgpstrModelIntegrationTest` (#22195) Update values Co-authored-by: ydshieh --- tests/models/mgp_str/test_modeling_mgp_str.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/mgp_str/test_modeling_mgp_str.py b/tests/models/mgp_str/test_modeling_mgp_str.py index ab5b6422b6..990f78587d 100644 --- a/tests/models/mgp_str/test_modeling_mgp_str.py +++ b/tests/models/mgp_str/test_modeling_mgp_str.py @@ -262,7 +262,7 @@ class MgpstrModelIntegrationTest(unittest.TestCase): self.assertEqual(out_strs["generated_text"][0], expected_text) expected_slice = torch.tensor( - [[[-39.7358, -44.8562, -36.6253], [-62.3605, -64.5908, -59.0069], [-74.6127, -68.9724, -71.7150]]], + [[[-39.5397, -44.4024, -36.1844], [-61.4709, -63.8639, -58.3454], [-74.0225, -68.5494, -71.2164]]], device=torch_device, )