From 60db26b0ac97d3aa4871bbeb0173932eea233ce3 Mon Sep 17 00:00:00 2001 From: Doug Turnbull Date: Tue, 7 May 2024 20:08:26 -0400 Subject: [PATCH] Remove memray test --- test/test_msmarco.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/test/test_msmarco.py b/test/test_msmarco.py index 3ad2dc2..c18a19d 100644 --- a/test/test_msmarco.py +++ b/test/test_msmarco.py @@ -185,15 +185,6 @@ def test_msmarco1m_phrase(phrase_search, msmarco1m, benchmark): profiler.run(msmarco1m['body_ws'].array.score, phrase_search) -@pytest.mark.parametrize("phrase_search", ["what is", "what is the", "what is the purpose", "what is the purpose of", "what is the purpose of cats", "star trek", "star trek the next generation", "what what what", "the purpose"]) -def test_msmarco1m_phrase_memray(phrase_search, msmarco1m, benchmark): - profiler = Profiler(benchmark) - phrase_search = phrase_search.split() - print(f"STARTING {phrase_search}") - print(f"Memory Usage (BODY): {msmarco1m['body_ws'].array.memory_usage() / 1024 ** 2:.2f} MB") - profiler.run(msmarco1m['body_ws'].array.score, phrase_search) - - @pytest.mark.skipif(not profile_enabled, reason="Profiling disabled") def test_msmarco10k_indexing(msmarco100k_raw, benchmark): profiler = Profiler(benchmark)