diff --git a/tests/test_function/test_search.py b/tests/test_function/test_search.py index c006b0d..995e3bc 100644 --- a/tests/test_function/test_search.py +++ b/tests/test_function/test_search.py @@ -4,13 +4,11 @@ import tempfile import logging import learnware - -learnware.init(logging_level=logging.WARNING) - from learnware.learnware import Learnware from learnware.client import LearnwareClient -from learnware.market import instantiate_learnware_market, BaseUserInfo, EasySemanticChecker -from learnware.config import C +from learnware.market import instantiate_learnware_market, BaseUserInfo + +learnware.init(logging_level=logging.WARNING) class TestSearch(unittest.TestCase): diff --git a/tests/test_workflow/test_hetero_workflow.py b/tests/test_workflow/test_hetero_workflow.py index efcc14c..117233f 100644 --- a/tests/test_workflow/test_hetero_workflow.py +++ b/tests/test_workflow/test_hetero_workflow.py @@ -11,9 +11,6 @@ from shutil import copyfile, rmtree from sklearn.metrics import mean_squared_error import learnware - -learnware.init(logging_level=logging.WARNING) - from learnware.market import instantiate_learnware_market, BaseUserInfo from learnware.specification import RKMETableSpecification, generate_rkme_table_spec, generate_semantic_spec from learnware.reuse import HeteroMapAlignLearnware, AveragingReuser, EnsemblePruningReuser @@ -21,7 +18,7 @@ from learnware.tests.templates import LearnwareTemplate, PickleModelTemplate, St from hetero_config import input_shape_list, input_description_list, output_description_list, user_description_list - +learnware.init(logging_level=logging.WARNING) curr_root = os.path.dirname(os.path.abspath(__file__)) diff --git a/tests/test_workflow/test_workflow.py b/tests/test_workflow/test_workflow.py index 0139b16..31656fe 100644 --- a/tests/test_workflow/test_workflow.py +++ b/tests/test_workflow/test_workflow.py @@ -10,14 +10,12 @@ from sklearn.datasets import load_digits from sklearn.model_selection import train_test_split import learnware - -learnware.init(logging_level=logging.WARNING) - from learnware.market import instantiate_learnware_market, BaseUserInfo from learnware.specification import RKMETableSpecification, generate_rkme_table_spec, generate_semantic_spec from learnware.reuse import JobSelectorReuser, AveragingReuser, EnsemblePruningReuser, FeatureAugmentReuser from learnware.tests.templates import LearnwareTemplate, PickleModelTemplate, StatSpecTemplate +learnware.init(logging_level=logging.WARNING) curr_root = os.path.dirname(os.path.abspath(__file__))