Browse Source

[MNT] Modify details

tags/v0.3.2
Gene 3 years ago
parent
commit
1de40299f1
2 changed files with 4 additions and 9 deletions
  1. +4
    -8
      examples/example_market_db/example_db.py
  2. +0
    -1
      learnware/market/easy.py

+ 4
- 8
examples/example_market_db/example_db.py View File

@@ -102,7 +102,6 @@ def test_market():
zip_path_list = get_zip_path_list() # the path list for learnware .zip

for idx, zip_path in enumerate(zip_path_list):
print(zip_path)
semantic_spec = semantic_specs[idx % 3]
semantic_spec["Name"]["Values"] = "learnware_%d" % (idx)
semantic_spec["Description"]["Values"] = "test_learnware_number_%d" % (idx)
@@ -146,8 +145,6 @@ def test_search_sementics():

os.system(f"rm -r {test_folder}")



def test_stat_search():
easy_market = EasyMarket()
@@ -164,7 +161,7 @@ def test_stat_search():
user_spec = specification.rkme.RKMEStatSpecification()
user_spec.load(os.path.join(unzip_dir, "svm.json"))
user_info = BaseUserInfo(
id="user_0", semantic_spec={"desc": "test_user_number_0"}, stat_info={"RKMEStatSpecification": user_spec}
id="user_0", semantic_spec=user_senmantic, stat_info={"RKMEStatSpecification": user_spec}
)
sorted_dist_list, single_learnware_list, mixture_learnware_list = easy_market.search_learnware(user_info)

@@ -179,8 +176,7 @@ def test_stat_search():

if __name__ == "__main__":
learnware_num = 10
# prepare_learnware(learnware_num)

# test_market()
# test_stat_search()
prepare_learnware(learnware_num)
test_market()
test_stat_search()
test_search_sementics()

+ 0
- 1
learnware/market/easy.py View File

@@ -368,7 +368,6 @@ class EasyMarket(BaseMarket):
learnware_list_tags = self._search_by_semantic_tags(learnware_list, user_info)
learnware_list_description = self._search_by_semantic_description(learnware_list, user_info)
learnware_list = list(set(learnware_list_tags + learnware_list_description))
# print(learnware_list_tags, learnware_list_description)

if "RKMEStatSpecification" not in user_info.stat_info:
return None, learnware_list, None


Loading…
Cancel
Save