Skip to content

Commit

Permalink
add comment
Browse files Browse the repository at this point in the history
  • Loading branch information
Samoed committed Sep 5, 2024
1 parent 98e85eb commit 12de698
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/test_load_results.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@

import pytest
from datasets import load_dataset
# for CI this should be `results.results`, for local just `results`
from results.results import MODELS

import mteb
Expand Down Expand Up @@ -31,6 +32,7 @@ def test_load_results():


@pytest.mark.parametrize("model", MODELS)
def test_load_results(model):
def test_load_results_from_datasets(model):
"""Ensures that all models can be imported from dataset"""
path = Path(__file__).parent.parent / "results.py"
ds = load_dataset(str(path.absolute()), model, trust_remote_code=True)

0 comments on commit 12de698

Please sign in to comment.