From df5b96d14bc1fce94b52ff183f0ff2d93055d503 Mon Sep 17 00:00:00 2001 From: Gene Date: Sun, 5 Nov 2023 21:33:02 +0800 Subject: [PATCH] [FIX] fix bugs about checker_names --- learnware/market/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/learnware/market/base.py b/learnware/market/base.py index 9d96f2c..9894c65 100644 --- a/learnware/market/base.py +++ b/learnware/market/base.py @@ -81,8 +81,6 @@ class LearnwareMarket: pending_learnware = get_learnware_from_dirpath( id="pending", semantic_spec=semantic_spec, learnware_dirpath=tempdir ) - checker_names = list(self.learnware_checker.keys()) if checker_names is None else checker_names - for name in checker_names: checker = self.learnware_checker[name] check_status = checker(pending_learnware) @@ -115,6 +113,7 @@ class LearnwareMarket: - str indicating model_id - int indicating the final learnware check_status """ + checker_names = list(self.learnware_checker.keys()) if checker_names is None else checker_names check_status = self.check_learnware(zip_path, semantic_spec, checker_names) return self.learnware_organizer.add_learnware( zip_path=zip_path, semantic_spec=semantic_spec, check_status=check_status, **kwargs @@ -178,6 +177,7 @@ class LearnwareMarket: if semantic_spec is None else semantic_spec ) + checker_names = list(self.learnware_checker.keys()) if checker_names is None else checker_names update_status = self.check_learnware(zip_path, semantic_spec, checker_names) check_status = ( update_status if check_status is None or update_status == BaseChecker.INVALID_LEARNWARE else check_status