From 4681a0a1a0a8230e2046ad0720e514e5c060574f Mon Sep 17 00:00:00 2001 From: Sebastian Schmidt Date: Wed, 2 Oct 2024 13:41:26 -0700 Subject: [PATCH] Update remaining data in text embedder test to match the output after XNNPack upgrade. PiperOrigin-RevId: 681576383 --- mediapipe/tasks/python/test/text/text_embedder_test.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mediapipe/tasks/python/test/text/text_embedder_test.py b/mediapipe/tasks/python/test/text/text_embedder_test.py index 028cc57d42..29ba13cced 100644 --- a/mediapipe/tasks/python/test/text/text_embedder_test.py +++ b/mediapipe/tasks/python/test/text/text_embedder_test.py @@ -110,7 +110,7 @@ def _check_cosine_similarity(self, result0, result1, expected_similarity): ModelFileType.FILE_NAME, 0.965793, 512, - (21.2054, 20.32263), + (21.2054, 19.6843), ), ( True, @@ -119,7 +119,7 @@ def _check_cosine_similarity(self, result0, result1, expected_similarity): ModelFileType.FILE_NAME, 0.965793, 512, - (0.062578, 0.065280), + (0.062578, 0.067393), ), ( False, @@ -203,7 +203,7 @@ def test_embed(self, l2_normalize, quantize, model_name, model_file_type, ModelFileType.FILE_NAME, 0.965793, 512, - (21.2054, 20.322639), + (21.2054, 19.684337), ), ( True, @@ -212,7 +212,7 @@ def test_embed(self, l2_normalize, quantize, model_name, model_file_type, ModelFileType.FILE_NAME, 0.965793, 512, - (0.0625787, 0.065280), + (0.0625787, 0.0673937), ), ( False,