Browse Source

[MNT] solve E402 error of flake8 in tests

tags/v0.3.2
Gene 1 year ago
parent
commit
173919e731
3 changed files with 5 additions and 12 deletions
  1. +3
    -5
      tests/test_function/test_search.py
  2. +1
    -4
      tests/test_workflow/test_hetero_workflow.py
  3. +1
    -3
      tests/test_workflow/test_workflow.py

+ 3
- 5
tests/test_function/test_search.py View File

@@ -4,13 +4,11 @@ import tempfile
import logging import logging


import learnware import learnware

learnware.init(logging_level=logging.WARNING)

from learnware.learnware import Learnware from learnware.learnware import Learnware
from learnware.client import LearnwareClient 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): class TestSearch(unittest.TestCase):


+ 1
- 4
tests/test_workflow/test_hetero_workflow.py View File

@@ -11,9 +11,6 @@ from shutil import copyfile, rmtree
from sklearn.metrics import mean_squared_error from sklearn.metrics import mean_squared_error


import learnware import learnware

learnware.init(logging_level=logging.WARNING)

from learnware.market import instantiate_learnware_market, BaseUserInfo from learnware.market import instantiate_learnware_market, BaseUserInfo
from learnware.specification import RKMETableSpecification, generate_rkme_table_spec, generate_semantic_spec from learnware.specification import RKMETableSpecification, generate_rkme_table_spec, generate_semantic_spec
from learnware.reuse import HeteroMapAlignLearnware, AveragingReuser, EnsemblePruningReuser 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 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__)) curr_root = os.path.dirname(os.path.abspath(__file__))






+ 1
- 3
tests/test_workflow/test_workflow.py View File

@@ -10,14 +10,12 @@ from sklearn.datasets import load_digits
from sklearn.model_selection import train_test_split from sklearn.model_selection import train_test_split


import learnware import learnware

learnware.init(logging_level=logging.WARNING)

from learnware.market import instantiate_learnware_market, BaseUserInfo from learnware.market import instantiate_learnware_market, BaseUserInfo
from learnware.specification import RKMETableSpecification, generate_rkme_table_spec, generate_semantic_spec from learnware.specification import RKMETableSpecification, generate_rkme_table_spec, generate_semantic_spec
from learnware.reuse import JobSelectorReuser, AveragingReuser, EnsemblePruningReuser, FeatureAugmentReuser from learnware.reuse import JobSelectorReuser, AveragingReuser, EnsemblePruningReuser, FeatureAugmentReuser
from learnware.tests.templates import LearnwareTemplate, PickleModelTemplate, StatSpecTemplate from learnware.tests.templates import LearnwareTemplate, PickleModelTemplate, StatSpecTemplate


learnware.init(logging_level=logging.WARNING)
curr_root = os.path.dirname(os.path.abspath(__file__)) curr_root = os.path.dirname(os.path.abspath(__file__))






Loading…
Cancel
Save