diff --git a/learnware/market/anchor.py b/learnware/market/anchor.py index 4500507..79d5443 100644 --- a/learnware/market/anchor.py +++ b/learnware/market/anchor.py @@ -51,8 +51,8 @@ class AnchoredMarket(BaseMarket): Basic market version """ - def __init__(self): - super(AnchoredMarket, self).__init__() + def __init__(self, *args, **kwargs): + super(AnchoredMarket, self).__init__(*args, **kwargs) self.anchor_learnware_list = {} # anchor_id: anchor learnware def _update_anchor_learnware(self, anchor_id: str, anchor_learnware: Learnware): diff --git a/learnware/market/evolve.py b/learnware/market/evolve.py index 7415de4..8912700 100644 --- a/learnware/market/evolve.py +++ b/learnware/market/evolve.py @@ -14,8 +14,8 @@ class EvolvedMarket(BaseMarket): Basic market version """ - def __init__(self): - super(EvolvedMarket, self).__init__() + def __init__(self, *args, **kwargs): + super(EvolvedMarket, self).__init__(*args, **kwargs) def generate_new_stat_specification(self, learnware: Learnware) -> BaseStatSpecification: """Generate new statistical specification for learnwares diff --git a/learnware/market/evolve_anchor.py b/learnware/market/evolve_anchor.py index adea95c..ef88479 100644 --- a/learnware/market/evolve_anchor.py +++ b/learnware/market/evolve_anchor.py @@ -15,8 +15,8 @@ class EvolvedAnchoredMarket(AnchoredMarket, EvolvedMarket): Market version with evolved learnwares """ - def __init__(self): - super(EvolvedAnchoredMarket, self).__init__() + def __init__(self, *args, **kwargs): + super(EvolvedAnchoredMarket, self).__init__(*args, **kwargs) def evolve_anchor_learnware_list(self, anchor_id_list: List[str]): """Enable anchor learnwares to evolve, e.g., new stat_spec diff --git a/learnware/market/heterogeneous_feature.py b/learnware/market/heterogeneous_feature.py index b25e2e8..8a611aa 100644 --- a/learnware/market/heterogeneous_feature.py +++ b/learnware/market/heterogeneous_feature.py @@ -34,8 +34,8 @@ class HeterogeneousFeatureMarket(EvolvedMarket): Market version with evolved learnwares """ - def __init__(self): - super(HeterogeneousFeatureMarket, self).__init__() + def __init__(self, *args, **kwargs): + super(HeterogeneousFeatureMarket, self).__init__(*args, **kwargs) self.mapping_function_list = {} def _mapping_function_list_initialization(self, learnware_list: List[Learnware]):