fix: change to relative import for tests

KennethEnevoldsen · Dec 7, 2023 · 3ec0673 · 3ec0673
1 parent 5e055da
commit 3ec0673
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/tests/test_all_models.py b/tests/test_all_models.py
@@ -4,7 +4,8 @@
 
 import pytest
 import seb
-from dummy_task import create_test_encode_task  # noqa: F401
+
+from .dummy_task import create_test_encode_task  # noqa: F401
 
 all_models = seb.get_all_models()
 openai_models = []

diff --git a/tests/test_benchmark.py b/tests/test_benchmark.py
@@ -9,9 +9,10 @@
 
 import pytest
 import seb
-from dummy_model import create_test_model  # noqa: F401
-from dummy_task import create_test_task  # noqa: F401
-from test_tasks import all_tasks_names
+
+from .dummy_model import create_test_model  # noqa: F401
+from .dummy_task import create_test_task  # noqa: F401
+from .test_tasks import all_tasks_names
 
 
 @pytest.mark.parametrize(