diff --git a/learnware/market/hetergeneous/organizer/__init__.py b/learnware/market/hetergeneous/organizer/__init__.py index c3c5d48..db0c05e 100644 --- a/learnware/market/hetergeneous/organizer/__init__.py +++ b/learnware/market/hetergeneous/organizer/__init__.py @@ -10,7 +10,8 @@ import pandas as pd from ....learnware import Learnware from ....logger import get_module_logger from ....specification.system import HeteroSpecification -from ...base import BaseOrganizer, BaseUserInfo +from ...base import BaseUserInfo +from ...easy2 import EasyOrganizer from ..database_ops import DatabaseOperations from .config import C as conf from .hetero_mapping import HeteroMapping, Trainer @@ -18,7 +19,7 @@ from .hetero_mapping import HeteroMapping, Trainer logger = get_module_logger("hetero_market") -class HeteroMapTableOrganizer(BaseOrganizer): +class HeteroMapTableOrganizer(EasyOrganizer): def reload_market(self, rebuild=False, auto_update_limit=50): self.market_store_path = os.path.join(conf.market_root_path, self.market_id) self.market_mapping_path = os.path.join(self.market_store_path, conf.market_model_path) @@ -29,9 +30,8 @@ class HeteroMapTableOrganizer(BaseOrganizer): self.learnware_zip_list = {} self.learnware_folder_list = {} self.count = 0 - # default root path: ../../.learnware - self.root_path = conf.market_root_path self.dbops = DatabaseOperations(conf.database_url, "market_" + self.market_id) + self.auto_update = False self.auto_update_limit = auto_update_limit os.makedirs(self.learnware_pool_path, exist_ok=True)