Browse Source

[MNT] modify details for check_user_info

tags/v0.3.2
Peng Tan 2 years ago
parent
commit
f694a6a8a1
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      learnware/market/heterogeneous/searcher.py

+ 2
- 2
learnware/market/heterogeneous/searcher.py View File

@@ -109,8 +109,8 @@ class HeteroSearcher(EasySearcher):
return False

user_input_description = user_info.get_semantic_spec()["Input"]
user_description_dim = int(user_input_description.get("Dimension", 0))
user_description_feature_num = len(user_input_description.get("Description", []))
user_description_dim = int(user_input_description["Dimension"])
user_description_feature_num = len(user_input_description["Description"])

if user_input_shape != user_description_dim or user_input_shape != user_description_feature_num:
logger.warning("User data feature dimensions mismatch with semantic specification.")


Loading…
Cancel
Save