From c63333d4c5db66112d2200ab95c88847fdec576b Mon Sep 17 00:00:00 2001 From: Gabriel Moreira Date: Wed, 13 Jul 2022 14:42:10 -0300 Subject: [PATCH] Fix retrieval test --- tests/unit/tf/models/test_retrieval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/tf/models/test_retrieval.py b/tests/unit/tf/models/test_retrieval.py index 3d25c6f758..5ebade4f7d 100644 --- a/tests/unit/tf/models/test_retrieval.py +++ b/tests/unit/tf/models/test_retrieval.py @@ -261,9 +261,9 @@ def test_two_tower_advanced_options(ecommerce_data): log_to_wandb=False, ) assert metrics["loss-final"] > 0.0 - assert metrics["recall_at_100-final"] > 0.0 assert metrics["runtime_sec-final"] > 0.0 assert metrics["avg_examples_per_sec-final"] > 0.0 + assert metrics["recall_at_10-final"] > 0.0 def test_mf_advanced_options(ecommerce_data): @@ -280,9 +280,9 @@ def test_mf_advanced_options(ecommerce_data): log_to_wandb=False, ) assert metrics["loss-final"] > 0.0 - assert metrics["recall_at_100-final"] > 0.0 assert metrics["runtime_sec-final"] > 0.0 assert metrics["avg_examples_per_sec-final"] > 0.0 + assert metrics["recall_at_10-final"] > 0.0 # def test_retrieval_evaluation_without_negatives(ecommerce_data: Dataset):