Browse Source

[FIX] BaseOrganizer -> EasyOrganizer

tags/v0.3.2
liuht 2 years ago
parent
commit
ed38d8afb6
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      learnware/market/hetergeneous/organizer/__init__.py

+ 4
- 4
learnware/market/hetergeneous/organizer/__init__.py View File

@@ -10,7 +10,8 @@ import pandas as pd
from ....learnware import Learnware from ....learnware import Learnware
from ....logger import get_module_logger from ....logger import get_module_logger
from ....specification.system import HeteroSpecification from ....specification.system import HeteroSpecification
from ...base import BaseOrganizer, BaseUserInfo
from ...base import BaseUserInfo
from ...easy2 import EasyOrganizer
from ..database_ops import DatabaseOperations from ..database_ops import DatabaseOperations
from .config import C as conf from .config import C as conf
from .hetero_mapping import HeteroMapping, Trainer from .hetero_mapping import HeteroMapping, Trainer
@@ -18,7 +19,7 @@ from .hetero_mapping import HeteroMapping, Trainer
logger = get_module_logger("hetero_market") logger = get_module_logger("hetero_market")




class HeteroMapTableOrganizer(BaseOrganizer):
class HeteroMapTableOrganizer(EasyOrganizer):
def reload_market(self, rebuild=False, auto_update_limit=50): 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_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) 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_zip_list = {}
self.learnware_folder_list = {} self.learnware_folder_list = {}
self.count = 0 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.dbops = DatabaseOperations(conf.database_url, "market_" + self.market_id)
self.auto_update = False
self.auto_update_limit = auto_update_limit self.auto_update_limit = auto_update_limit


os.makedirs(self.learnware_pool_path, exist_ok=True) os.makedirs(self.learnware_pool_path, exist_ok=True)


Loading…
Cancel
Save