From f6e5388e24e9865bf8be0538e44290da6cf9adbe Mon Sep 17 00:00:00 2001 From: elronbandel Date: Thu, 30 Jan 2025 11:22:35 +0200 Subject: [PATCH] Refactor translation card testing logic and remove unused dataset loading Signed-off-by: elronbandel --- prepare/cards/translation/flores101.py | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/prepare/cards/translation/flores101.py b/prepare/cards/translation/flores101.py index cf00e1f04..e2d96264c 100644 --- a/prepare/cards/translation/flores101.py +++ b/prepare/cards/translation/flores101.py @@ -150,17 +150,8 @@ task="tasks.translation.directed", templates="templates.translation.directed.all", ) - - test_card(card, demos_taken_from="test") + if pair == pairs[0]: + test_card(card, demos_taken_from="test") add_to_catalog( card, f"cards.mt.flores_101.{pair['src']}_{pair['tgt']}", overwrite=True ) - -if __name__ == "__main__": - from unitxt import load_dataset - - ds = load_dataset( - "card=cards.mt.flores_101.eng_deu,template_card_index=0", - ) - - ds["test"][0]