From f0b395912a8175b1dd9df9a80427a660e9db2ebf Mon Sep 17 00:00:00 2001 From: Francisco Aranda Date: Mon, 28 Feb 2022 14:13:55 +0100 Subject: [PATCH] fix: update dataset name --- tests/functional_tests/test_log_for_token_classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/functional_tests/test_log_for_token_classification.py b/tests/functional_tests/test_log_for_token_classification.py index 8e49d688a9..658744c8bd 100644 --- a/tests/functional_tests/test_log_for_token_classification.py +++ b/tests/functional_tests/test_log_for_token_classification.py @@ -450,7 +450,7 @@ def test_search_keywords(mocked_client): dataset = "test_search_keywords" from datasets import load_dataset - dataset_ds = load_dataset("rubrix/gutenberg_spacy-ner_sm", split="train") + dataset_ds = load_dataset("rubrix/gutenberg_spacy-ner", split="train") dataset_rb = rubrix.read_datasets(dataset_ds, task="TokenClassification") rubrix.delete(dataset)