fixes for imports

rnugmanx · Jan 22, 2024 · b144568 · b144568
1 parent 2ea2488
commit b144568
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/tests/e2e_oss/collect_refs.py b/tests/e2e_oss/collect_refs.py
@@ -17,7 +17,7 @@
 from e2e_oss.common.parsers import pipeline_cfg_to_string
 from e2e_oss.common.common.pipeline import Pipeline
 
-pytest_plugins = ('e2e_oss.plugins.ref_collect.conftest', )
+pytest_plugins = ('e2e_oss.common.plugins.ref_collect.conftest', )
 
 
 def save_reference(refs, path, use_torch_to_save):

diff --git a/tests/e2e_oss/common/plugins/e2e_test/conftest.py b/tests/e2e_oss/common/plugins/e2e_test/conftest.py
@@ -15,7 +15,7 @@
 import yaml
 from _pytest.runner import show_test_item, call_runtest_hook, check_interactive_exception
 
-import e2e_oss.plugins.common.base_conftest as base
+import e2e_oss.common.plugins.common.base_conftest as base
 
 from e2e_oss.test_utils.path_utils import DirLockingHandler
 from e2e_oss.test_utils.test_utils import class_factory, BrokenTest, BrokenTestException

diff --git a/tests/e2e_oss/common/plugins/ref_collect/conftest.py b/tests/e2e_oss/common/plugins/ref_collect/conftest.py
@@ -3,7 +3,7 @@
 
 import yaml
 
-import e2e_oss.plugins.common.base_conftest as base
+import e2e_oss.common.plugins.common.base_conftest as base
 
 from e2e_oss.common.env_utils import fix_env_conf
 from e2e_oss.test_utils.env_tools import Environment

diff --git a/tests/e2e_oss/common/plugins/reshape_tests/conftest.py b/tests/e2e_oss/common/plugins/reshape_tests/conftest.py
@@ -11,12 +11,12 @@
 import pytest
 import yaml
 
-import e2e_oss.plugins.common.base_conftest as base
+import e2e_oss.common.plugins.common.base_conftest as base
 from e2e_oss.test_utils.reshape_tests_utils import should_run_reshape, get_reshape_configurations, \
     get_reshape_pipeline_pairs, batch_was_changed
 from e2e_oss.test_utils.test_utils import class_factory, BrokenTest, BrokenTestException
 from e2e_oss.common.env_utils import fix_env_conf
-from e2e_oss.plugins.e2e_test.conftest import satisfies_all_rules, unwrap_rules
+from e2e_oss.common.plugins.e2e_test.conftest import satisfies_all_rules, unwrap_rules
 from e2e_oss.test_utils.env_tools import Environment
 
 

diff --git a/tests/e2e_oss/test_utils/help_scripts/get_tests_name.py b/tests/e2e_oss/test_utils/help_scripts/get_tests_name.py
@@ -12,7 +12,7 @@
 
 import os
 import pytest
-pytest_plugins = ('e2e_oss.plugins.e2e_test.conftest',)
+pytest_plugins = ('e2e_oss.common.plugins.e2e_test.conftest',)
 
 
 @pytest.fixture(scope="session")