From a25875a34d8a1af5ae25e2023b388dd538b3db4d Mon Sep 17 00:00:00 2001 From: Arne Binder Date: Tue, 24 Oct 2023 20:55:09 +0200 Subject: [PATCH] use Tacred.BASE_DATASET_PATH --- tests/dataset_builders/pie/test_tacred.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/dataset_builders/pie/test_tacred.py b/tests/dataset_builders/pie/test_tacred.py index 4c0accb5..8a9c7217 100644 --- a/tests/dataset_builders/pie/test_tacred.py +++ b/tests/dataset_builders/pie/test_tacred.py @@ -22,8 +22,8 @@ logger = logging.getLogger(__name__) -HF_DATASET_PATH = "DFKI-SLT/tacred" PIE_DATASET_PATH = f"{PIE_BASE_PATH}/tacred" +HF_DATASET_PATH = Tacred.BASE_DATASET_PATH SPLITS = ["train", "validation", "test"] EXAMPLE_IDX = 0 NUM_SAMPLES = 3