Skip to content

Commit

Permalink
Update remaining data in text embedder test to match the output after…
Browse files Browse the repository at this point in the history
… XNNPack upgrade.

PiperOrigin-RevId: 681576383
  • Loading branch information
schmidt-sebastian authored and copybara-github committed Oct 2, 2024
1 parent 0e634ea commit 4681a0a
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions mediapipe/tasks/python/test/text/text_embedder_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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,
Expand Down Expand Up @@ -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,
Expand All @@ -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,
Expand Down

0 comments on commit 4681a0a

Please sign in to comment.