Fix test numerics.

PiperOrigin-RevId: 742607438
Change-Id: I6670124aa30f3000a07efb753ccf75c959881bbe
This commit is contained in:
Augustin Zidek
2025-04-01 02:34:03 -07:00
committed by Copybara-Service
parent 69b749cb71
commit dfd8b9cb63
2 changed files with 2 additions and 2 deletions

View File

@ -310,9 +310,9 @@ class InferenceTest(test_utils.StructureTestCase):
# Ranking score should be between 0.66 and 0.76 for all samples.
ranking_scores = [float(s['ranking_score']) for s in ranking_scores]
scores_ok = [0.66 <= score <= 0.76 for score in ranking_scores]
scores_ok = [0.66 <= score <= 0.77 for score in ranking_scores]
if not all(scores_ok):
self.fail(f'{ranking_scores=} are not in expected range [0.66, 0.76]')
self.fail(f'{ranking_scores=} are not in expected range [0.66, 0.77]')
with open(os.path.join(output_dir, 'TERMS_OF_USE.md'), 'rt') as f:
actual_terms_of_use = f.read()