Fix tasks unit test includes

timofurrer · Dec 20, 2019 · 6767d61 · 6767d61
1 parent 5fa45da
commit 6767d61
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/tasks/test_odd_one_out.py b/tests/tasks/test_odd_one_out.py
@@ -8,7 +8,7 @@
 :license: MIT, see LICENSE for more details.
 """
 
-from embedeval.tasks.odd_one_out import OddOneOutTask
+from embedeval.tasks.en_got_odd_one_out import OddOneOutTask
 from unittest.mock import MagicMock
 
 

diff --git a/tests/tasks/test_offense_detection.py b/tests/tasks/test_offense_detection.py
@@ -10,7 +10,7 @@
 
 import pytest
 
-from embedeval.tasks.offense_detection import OffenseDetectionTask
+from embedeval.tasks.de_offense_detection import OffenseDetectionTask
 
 
 @pytest.fixture(name="offense_detection_task_setup")

diff --git a/tests/tasks/test_word_analogy.py b/tests/tasks/test_word_analogy.py
@@ -8,7 +8,7 @@
 :license: MIT, see LICENSE for more details.
 """
 
-from embedeval.tasks.word_analogy import WordAnalogyTask
+from embedeval.tasks.en_got_word_analogy import WordAnalogyTask
 from unittest.mock import MagicMock