Browse Source

[MNT] enhance check_user_info and verify it with test

tags/v0.3.2
Peng Tan 2 years ago
parent
commit
46d84c1f72
2 changed files with 6 additions and 1 deletions
  1. +5
    -0
      learnware/market/heterogeneous/searcher.py
  2. +1
    -1
      learnware/market/module.py

+ 5
- 0
learnware/market/heterogeneous/searcher.py View File

@@ -105,6 +105,11 @@ class HeteroSearcher(EasySearcher):

user_input_description = user_info.get_semantic_spec()["Input"]

user_task_type=user_info.get_semantic_spec()["Task"]["Values"]
if user_task_type not in [["Classification"], ["Regression"]]:
logger.warning("User doesn't provide correct task type, it must be either Classification or Regression")
return False

user_description_dim = int(user_input_description["Dimension"])
user_description_feature_num = len(user_input_description["Description"])



+ 1
- 1
learnware/market/module.py View File

@@ -11,7 +11,7 @@ MARKET_CONFIG = {
"hetero": {
"organizer": HeteroMapTableOrganizer(),
"searcher": HeteroSearcher(),
"checker_list": [EasySemanticChecker(), EasyStatChecker()]
"checker_list": []
}
}



Loading…
Cancel
Save