From 02f6c0c6b8f4029cbae978550c67a03bb0e32c74 Mon Sep 17 00:00:00 2001 From: 51616 Date: Tue, 18 Feb 2025 08:11:08 +0000 Subject: [PATCH] add squad test --- src/ctx_to_lora/data_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/ctx_to_lora/data_utils.py b/src/ctx_to_lora/data_utils.py index c7387f2..b25b377 100644 --- a/src/ctx_to_lora/data_utils.py +++ b/src/ctx_to_lora/data_utils.py @@ -36,6 +36,7 @@ DS_KWARGS = { "squad": dict( train=dict(path="rajpurkar/squad", split="train[900:]"), validation=dict(path="rajpurkar/squad", split="train[:900]"), + test=dict(path="rajpurkar/squad", split="validation"), ), "fw_qa_tiny": dict( train=dict(