Skip to content

Commit

Permalink
Remove memray test
Browse files Browse the repository at this point in the history
  • Loading branch information
softwaredoug committed May 8, 2024
1 parent a2f74ab commit 60db26b
Showing 1 changed file with 0 additions and 9 deletions.
9 changes: 0 additions & 9 deletions test/test_msmarco.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 60db26b

Please sign in to comment.