From 344f5269b293af3e07150ab4767dd57f47229383 Mon Sep 17 00:00:00 2001 From: Vinnam Kim Date: Mon, 10 Jul 2023 15:25:14 +0900 Subject: [PATCH] Fix import dm.DatasetSubset (#2324) Signed-off-by: Kim, Vinnam --- tests/unit/core/data/manager/test_dataset_manager.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit/core/data/manager/test_dataset_manager.py b/tests/unit/core/data/manager/test_dataset_manager.py index ba1daa768f9..41a84740d95 100644 --- a/tests/unit/core/data/manager/test_dataset_manager.py +++ b/tests/unit/core/data/manager/test_dataset_manager.py @@ -6,9 +6,9 @@ from typing import List from tempfile import TemporaryDirectory -import datumaro as dm import pytest +from datumaro.components.dataset import DatasetSubset from otx.cli.manager.config_manager import TASK_TYPE_TO_SUPPORTED_FORMAT from otx.core.data.manager.dataset_manager import DatasetManager from tests.test_suite.e2e_test_system import e2e_pytest_unit @@ -51,7 +51,7 @@ def test_get_train_dataset(self, task: List[str], subset: List[str]): DatasetManager.get_train_dataset(self.dataset[subset][task]) else: train_dataset = DatasetManager.get_train_dataset(self.dataset[subset][task]) - assert isinstance(train_dataset, dm.DatasetSubset) + assert isinstance(train_dataset, DatasetSubset) @e2e_pytest_unit @pytest.mark.parametrize("task", AVAILABLE_TASKS) @@ -61,7 +61,7 @@ def test_get_val_dataset(self, task: List[str], subset: List[str]): assert DatasetManager.get_val_dataset(self.dataset[subset][task]) is None else: val_dataset = DatasetManager.get_val_dataset(self.dataset[subset][task]) - assert isinstance(val_dataset, dm.DatasetSubset) + assert isinstance(val_dataset, DatasetSubset) @e2e_pytest_unit @pytest.mark.parametrize("data_root", AVAILABLE_DATA_ROOTS)