diff --git a/tests/e2e/cli/semantic_segmentation/test_segmentation.py b/tests/e2e/cli/semantic_segmentation/test_segmentation.py index 3bfd0af3a8e..0315e3e051d 100644 --- a/tests/e2e/cli/semantic_segmentation/test_segmentation.py +++ b/tests/e2e/cli/semantic_segmentation/test_segmentation.py @@ -318,8 +318,6 @@ class TestToolsOTXSelfSLSegmentation: @pytest.mark.parametrize("template", templates, ids=templates_ids) def test_otx_train(self, template, tmp_dir_path): tmp_dir_path_1 = tmp_dir_path / "segmentation/test_selfsl" - if not (Path(template.model_template_path).parent / "selfsl").is_dir(): - pytest.skip("Self-SL training type isn't available for this template") otx_train_testing(template, tmp_dir_path_1, otx_dir, args_selfsl) template_work_dir = get_template_dir(template, tmp_dir_path_1) assert (Path(template_work_dir) / "selfsl").is_dir() @@ -332,8 +330,6 @@ def test_otx_train(self, template, tmp_dir_path): @pytest.mark.skipif(TT_STABILITY_TESTS, reason="This is TT_STABILITY_TESTS") @pytest.mark.parametrize("template", templates, ids=templates_ids) def test_otx_eval(self, template, tmp_dir_path): - if not (Path(template.model_template_path).parent / "selfsl").is_dir(): - pytest.skip("Self-SL training type isn't available for this template") tmp_dir_path = tmp_dir_path / "segmentation/test_selfsl_sl" otx_eval_testing(template, tmp_dir_path, otx_dir, args) @@ -342,8 +338,6 @@ def test_otx_eval(self, template, tmp_dir_path): @pytest.mark.skipif(MULTI_GPU_UNAVAILABLE, reason="The number of gpu is insufficient") @pytest.mark.parametrize("template", templates, ids=templates_ids) def test_otx_multi_gpu_train_selfsl(self, template, tmp_dir_path): - if not (Path(template.model_template_path).parent / "selfsl").is_dir(): - pytest.skip("Self-SL training type isn't available for this template") tmp_dir_path = tmp_dir_path / "segmentation/test_multi_gpu_selfsl" args_selfsl_multigpu = copy.deepcopy(args_selfsl) args_selfsl_multigpu["--gpus"] = "0,1" diff --git a/tests/integration/cli/semantic_segmentation/test_segmentation.py b/tests/integration/cli/semantic_segmentation/test_segmentation.py index d47166f0170..0221b1c375a 100644 --- a/tests/integration/cli/semantic_segmentation/test_segmentation.py +++ b/tests/integration/cli/semantic_segmentation/test_segmentation.py @@ -203,8 +203,6 @@ def test_otx_multi_gpu_train_semisl(self, template, tmp_dir_path): @e2e_pytest_component @pytest.mark.parametrize("template", default_templates, ids=default_templates_ids) def test_otx_train_selfsl(self, template, tmp_dir_path): - if not (Path(template.model_template_path).parent / "selfsl").is_dir(): - pytest.skip("Self-SL training type isn't available for this template") tmp_dir_path = tmp_dir_path / "segmentation/test_selfsl" otx_train_testing(template, tmp_dir_path, otx_dir, args_selfsl) @@ -212,8 +210,6 @@ def test_otx_train_selfsl(self, template, tmp_dir_path): @pytest.mark.skipif(MULTI_GPU_UNAVAILABLE, reason="The number of gpu is insufficient") @pytest.mark.parametrize("template", default_templates, ids=default_templates_ids) def test_otx_multi_gpu_train_selfsl(self, template, tmp_dir_path): - if not (Path(template.model_template_path).parent / "selfsl").is_dir(): - pytest.skip("Self-SL training type isn't available for this template") tmp_dir_path = tmp_dir_path / "segmentation/test_multi_gpu_selfsl" args_selfsl_multigpu = copy.deepcopy(args_selfsl) args_selfsl_multigpu["--gpus"] = "0,1" diff --git a/tests/regression/semantic_segmentation/test_segmentation.py b/tests/regression/semantic_segmentation/test_segmentation.py index 2505aff3ba2..91a48ec3fee 100644 --- a/tests/regression/semantic_segmentation/test_segmentation.py +++ b/tests/regression/semantic_segmentation/test_segmentation.py @@ -244,8 +244,6 @@ def test_otx_train_semisl_kpi_test(self, reg_cfg, template): @e2e_pytest_component @pytest.mark.parametrize("template", templates, ids=templates_ids) def test_otx_train_selfsl(self, reg_cfg, template, tmp_dir_path): - if not (Path(template.model_template_path).parent / "selfsl").is_dir(): - pytest.skip("Self-SL training type isn't available for this template") train_type = "self_supervised" self.performance[template.name] = {} @@ -297,8 +295,6 @@ def test_otx_train_selfsl(self, reg_cfg, template, tmp_dir_path): @e2e_pytest_component @pytest.mark.parametrize("template", templates, ids=templates_ids) def test_otx_train_selfsl_kpi_test(self, reg_cfg, template): - if not (Path(template.model_template_path).parent / "selfsl").is_dir(): - pytest.skip("Self-SL training type isn't available for this template") train_type = "self_supervised" config_selfsl = reg_cfg.load_config(train_type=train_type) performance = reg_cfg.get_template_performance(template, train_type=train_type)