Skip to content

Commit

Permalink
Update path for CI tests (#1153)
Browse files Browse the repository at this point in the history
  • Loading branch information
harimkang authored Jul 7, 2022
1 parent b8722fc commit 29ea726
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions external/model-preparation-algorithm/tests/test_ote_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,9 +55,9 @@
from ote_sdk.usecases.tasks.interfaces.export_interface import ExportType
from ote_sdk.utils.shape_factory import ShapeFactory

DEFAULT_CLS_TEMPLATE_DIR = osp.join('external', 'model-preparation-algorithm', 'configs', 'classification', 'efficientnet_b0_cls_incr')
DEFAULT_DET_TEMPLATE_DIR = osp.join('external', 'model-preparation-algorithm', 'configs', 'detection', 'mobilenetv2_atss_cls_incr')
DEFAULT_SEG_TEMPLATE_DIR = osp.join('external', 'model-preparation-algorithm', 'configs', 'segmentation', 'ocr-lite-hrnet-18-cls-incr')
DEFAULT_CLS_TEMPLATE_DIR = osp.join('configs', 'classification', 'efficientnet_b0_cls_incr')
DEFAULT_DET_TEMPLATE_DIR = osp.join('configs', 'detection', 'mobilenetv2_atss_cls_incr')
DEFAULT_SEG_TEMPLATE_DIR = osp.join('configs', 'segmentation', 'ocr-lite-hrnet-18-cls-incr')


def eval(task: BaseTask, model: ModelEntity, dataset: DatasetEntity) -> Performance:
Expand All @@ -82,7 +82,7 @@ def test_reading_classification_cls_incr_model_template(self):
'mobilenet_v3_large_1_cls_incr', 'mobilenet_v3_large_075_cls_incr',
'mobilenet_v3_small_cls_incr']
for model_template in classification_template:
parse_model_template(osp.join('external', 'model-preparation-algorithm', 'configs', 'classification', model_template, 'template_experimental.yaml'))
parse_model_template(osp.join('configs', 'classification', model_template, 'template_experimental.yaml'))

@staticmethod
def generate_label_schema(not_empty_labels, multilabel=False):
Expand Down Expand Up @@ -239,7 +239,7 @@ class TestMPADetAPI:
def test_reading_detection_cls_incr_model_template(self):
detection_template = ['mobilenetv2_atss_cls_incr', 'resnet50_vfnet_cls_incr']
for model_template in detection_template:
parse_model_template(osp.join('external', 'model-preparation-algorithm', 'configs', 'detection', model_template, 'template_experimental.yaml'))
parse_model_template(osp.join('configs', 'detection', model_template, 'template_experimental.yaml'))

def init_environment(
self,
Expand Down Expand Up @@ -452,7 +452,7 @@ class TestMPASegAPI:
def test_reading_segmentation_cls_incr_model_template(self):
segmentation_template = ['ocr-lite-hrnet-18-cls-incr']
for model_template in segmentation_template:
parse_model_template(osp.join('external', 'model-preparation-algorithm', 'configs', 'segmentation', model_template, 'template_experimental.yaml'))
parse_model_template(osp.join('configs', 'segmentation', model_template, 'template_experimental.yaml'))

@staticmethod
def generate_label_schema(label_names):
Expand Down

0 comments on commit 29ea726

Please sign in to comment.