GH-243: fix test

flairNLP · Nov 26, 2018 · b12b073 · b12b073
1 parent 2d819e4
commit b12b073
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_hyperparameter.py b/tests/test_hyperparameter.py
@@ -12,7 +12,7 @@
 
 @pytest.mark.integration
 def test_sequence_tagger_param_selector(results_base_path, tasks_base_path):
-    corpus = NLPTaskDataFetcher.fetch_data(NLPTask.FASHION, base_path=tasks_base_path)
+    corpus = NLPTaskDataFetcher.load_corpus(NLPTask.FASHION, base_path=tasks_base_path)
 
     # define search space
     search_space = SearchSpace()
@@ -49,7 +49,7 @@ def test_sequence_tagger_param_selector(results_base_path, tasks_base_path):
 
 @pytest.mark.integration
 def test_text_classifier_param_selector(results_base_path, tasks_base_path):
-    corpus = NLPTaskDataFetcher.fetch_data(NLPTask.IMDB, base_path=tasks_base_path)
+    corpus = NLPTaskDataFetcher.load_corpus(NLPTask.IMDB, base_path=tasks_base_path)
 
     glove_embedding: WordEmbeddings = WordEmbeddings('en-glove')