Skip to content

Commit

Permalink
Remove skip for segnext
Browse files Browse the repository at this point in the history
  • Loading branch information
sungchul2 committed Sep 15, 2023
1 parent c04ec31 commit 7225d02
Show file tree
Hide file tree
Showing 3 changed files with 0 additions and 14 deletions.
6 changes: 0 additions & 6 deletions tests/e2e/cli/semantic_segmentation/test_segmentation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand All @@ -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)

Expand All @@ -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"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -203,17 +203,13 @@ 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)

@e2e_pytest_component
@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"
Expand Down
4 changes: 0 additions & 4 deletions tests/regression/semantic_segmentation/test_segmentation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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] = {}

Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit 7225d02

Please sign in to comment.