From 18eb112f48808f4a1b44b60c1c404cced7692124 Mon Sep 17 00:00:00 2001 From: zouxiaochuan Date: Thu, 20 Jul 2023 16:39:47 +0800 Subject: [PATCH] [MNT] remove check keys in semantic spec search --- learnware/market/easy.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/learnware/market/easy.py b/learnware/market/easy.py index 5f504e8..d6af5f5 100644 --- a/learnware/market/easy.py +++ b/learnware/market/easy.py @@ -585,10 +585,14 @@ class EasyMarket(BaseMarket): def _search_by_semantic_spec(self, learnware_list: List[Learnware], user_info: BaseUserInfo) -> List[Learnware]: def match_semantic_spec(semantic_spec1, semantic_spec2): + ''' + semantic_spec1: semantic spec input by user + semantic_spec2: semantic spec in database + ''' if semantic_spec1.keys() != semantic_spec2.keys(): - # raise Exception("semantic_spec key error") - logger.warning("semantic_spec key error!") - return False + # sematic spec in database may contain more keys than user input + pass + for key in semantic_spec1.keys(): if len(semantic_spec1[key]["Values"]) == 0: continue @@ -618,7 +622,7 @@ class EasyMarket(BaseMarket): for learnware in learnware_list: learnware_semantic_spec = learnware.get_specification().get_semantic_spec() user_semantic_spec = user_info.get_semantic_spec() - if match_semantic_spec(learnware_semantic_spec, user_semantic_spec): + if match_semantic_spec(user_semantic_spec, learnware_semantic_spec): match_learnwares.append(learnware) logger.info("semantic_spec search: choose %d from %d learnwares" % (len(match_learnwares), len(learnware_list))) return match_learnwares