diff --git a/tests/test_encoded_queries.py b/tests/test_encoded_queries.py index 663def643..a006e458f 100644 --- a/tests/test_encoded_queries.py +++ b/tests/test_encoded_queries.py @@ -46,7 +46,7 @@ def test_tct_colbert_msmarco_doc_dev(self): def test_ance_maxp_msmarco_doc_dev(self): encoder = QueryEncoder.load_encoded_queries('ance_maxp-msmarco-doc-dev') - topics = get_topics('maxp-msmarco-doc-dev') + topics = get_topics('msmarco-doc-dev') for t in topics: self.assertTrue(topics[t]['title'] in encoder.embedding)