From 04bf37452ff19f668efe9676180e435c8c75c0ef Mon Sep 17 00:00:00 2001 From: Gene Date: Mon, 13 Nov 2023 22:42:05 +0800 Subject: [PATCH] [FIX] add None check in load market --- learnware/market/easy/database_ops.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/learnware/market/easy/database_ops.py b/learnware/market/easy/database_ops.py index a0b163c..5eed22d 100644 --- a/learnware/market/easy/database_ops.py +++ b/learnware/market/easy/database_ops.py @@ -203,12 +203,17 @@ class DatabaseOperations(object): for id, semantic_spec, zip_path, folder_path, use_flag in cursor: id = id.strip() semantic_spec_dict = json.loads(semantic_spec) - new_learnware = get_learnware_from_dirpath( - id=id, semantic_spec=semantic_spec_dict, learnware_dirpath=folder_path - ) - logger.info(f"Load learnware: {id}") + try: + new_learnware = get_learnware_from_dirpath( + id=id, semantic_spec=semantic_spec_dict, learnware_dirpath=folder_path + ) + assert new_learnware is not None, "learnware must not be None" + logger.info(f"Load learnware: {id}") + except Exception as err: + logger.error(f"Load learnware {id} failed due to {err}!") + continue + learnware_list[id] = new_learnware - # assert new_learnware is not None zip_list[id] = zip_path folder_list[id] = folder_path use_flags[id] = int(use_flag)