Skip to content

Commit

Permalink
add verification of hard coded splits into test_card
Browse files Browse the repository at this point in the history
Signed-off-by: dafnapension <[email protected]>
  • Loading branch information
dafnapension committed Feb 8, 2025
1 parent 4ba542d commit 68f7b58
Showing 1 changed file with 19 additions and 0 deletions.
19 changes: 19 additions & 0 deletions src/unitxt/test_utils/card.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
from .. import add_to_catalog, register_local_catalog
from ..artifact import fetch_artifact
from ..collections import Collection
from ..loaders import LoadHF
from ..logging_utils import get_logger
from ..metric import _compute
from ..settings_utils import get_settings
Expand Down Expand Up @@ -59,6 +60,24 @@ def load_examples_from_dataset_recipe(card, template_card_index, debug, **kwargs
kwargs["template_card_index"] = template_card_index

recipe = DatasetRecipe(card=card, **kwargs)
loader = recipe.loading.steps[0]
if (
isinstance(loader, LoadHF)
and hasattr(loader, "splits")
and loader.splits is not None
and isinstance(loader.splits, list)
):
endms = recipe()
keys_endms = str(sorted(endms.keys()))
saved_splits = loader.splits
loader.splits = None
non_splits_endms = recipe()
keys_non_splits_endms = str(sorted(non_splits_endms.keys()))
loader.splits = saved_splits
assert (
keys_endms == keys_non_splits_endms
), f"splits names are not as expected for card of path {loader.path}"

logger.info(f"Using these card recipe parameters: {kwargs}")

if debug:
Expand Down

0 comments on commit 68f7b58

Please sign in to comment.