-
Notifications
You must be signed in to change notification settings - Fork 461
[v2] Merge MIEB
into v2
#1973
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[v2] Merge MIEB
into v2
#1973
Conversation
* Fix training data * Fix data * Fix data --------- Co-authored-by: Kenneth Enevoldsen <kennethcenevoldsen@gmail.com>
Update SCIDOCSNLRetrieval.py
* add model memory usage * lint * update * calculate memory usage based on file size * calculate memory usage * add memory usage for MIEB models * add last model usage * add memory_usage_mb to overview * fix rerank * update memory usage * update memory usage
# Conflicts: # mteb/model_meta.py # mteb/models/gritlm_models.py # mteb/models/overview.py # mteb/models/promptriever_models.py # mteb/tasks/Retrieval/eng/HagridRetrieval.py
# Conflicts: # tests/test_tasks/test_all_abstasks.py
# Conflicts: # mteb/benchmarks/benchmarks.py # mteb/tasks/__init__.py # mteb/tasks/aggregated_tasks/__init__.py
# Conflicts: # mteb/model_meta.py # mteb/models/bge_models.py # mteb/models/cde_models.py # mteb/models/cohere_models.py # mteb/models/colbert_models.py # mteb/models/e5_instruct.py # mteb/models/google_models.py # mteb/models/gritlm_models.py # mteb/models/jasper_models.py # mteb/models/jina_models.py # mteb/models/lens_models.py # mteb/models/llm2vec_models.py # mteb/models/misc_models.py # mteb/models/model2vec_models.py # mteb/models/no_instruct_sentence_models.py # mteb/models/nomic_models.py # mteb/models/openai_models.py # mteb/models/promptriever_models.py # mteb/models/repllama_models.py # mteb/models/ru_sentence_models.py # mteb/models/salesforce_models.py # mteb/models/uae_models.py # mteb/models/voyage_models.py
@isaac-chung Can you review PR? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Wow nice! Just got a few small questions.
@@ -7,6 +7,7 @@ | |||
|
|||
@pytest.mark.parametrize("task", MOCK_TASK_TEST_GRID) | |||
def test_descriptive_stats(task): | |||
# todo integrate MIEB |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Is this meant to be done in this PR?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
No. Not all MIEB
tasks have implemented calculate_descripteve_stats
this test should be updated after it
Code Quality
make lint
to maintain consistent style.Documentation
Testing
make test-with-coverage
.make test
ormake test-with-coverage
to ensure no existing functionality is broken.