Skip to content
Snippets Groups Projects
Commit e8e9b289 authored by Stefano Borini's avatar Stefano Borini
Browse files

Reworked plugins to honor the new interface

parent 82354b1a
No related branches found
No related tags found
1 merge request!29Extract io layer to writer/reader class
...@@ -12,9 +12,9 @@ class CSVExtractorBundle(HasStrictTraits): ...@@ -12,9 +12,9 @@ class CSVExtractorBundle(HasStrictTraits):
def create_model(self, model_data=None): def create_model(self, model_data=None):
if model_data is None: if model_data is None:
return CSVExtractorModel() model_data = {}
else:
return CSVExtractorModel.from_json(model_data) return CSVExtractorModel(self, **model_data)
def create_data_source(self, application, model): def create_data_source(self, application, model):
return CSVExtractorDataSource(self, application, model) return CSVExtractorDataSource(self, application, model)
...@@ -13,9 +13,9 @@ class KPIAdderBundle(HasStrictTraits): ...@@ -13,9 +13,9 @@ class KPIAdderBundle(HasStrictTraits):
def create_model(self, model_data=None): def create_model(self, model_data=None):
if model_data is None: if model_data is None:
return KPIAdderModel() model_data = {}
else:
return KPIAdderModel.from_json(model_data) return KPIAdderModel(self, **model_data)
def create_data_source(self, application, model): def create_data_source(self, application, model):
return KPIAdderCalculator(self, application, model) return KPIAdderCalculator(self, application, model)
...@@ -12,9 +12,8 @@ class DakotaBundle(HasStrictTraits): ...@@ -12,9 +12,8 @@ class DakotaBundle(HasStrictTraits):
def create_model(self, model_data=None): def create_model(self, model_data=None):
if model_data is None: if model_data is None:
return DakotaModel() model_data = {}
else: return DakotaModel(self, **model_data)
return DakotaModel.from_json(model_data)
def create_optimizer(self, application, model): def create_optimizer(self, application, model):
return DakotaOptimizer(self, application, model) return DakotaOptimizer(self, application, model)
......
...@@ -8,4 +8,4 @@ class BaseDataSourceModel(ABCHasStrictTraits): ...@@ -8,4 +8,4 @@ class BaseDataSourceModel(ABCHasStrictTraits):
def __init__(self, bundle, *args, **kwargs): def __init__(self, bundle, *args, **kwargs):
self.bundle = bundle self.bundle = bundle
super(BaseDataSourceModel, self).__init__(self, *args, **kwargs) super(BaseDataSourceModel, self).__init__(*args, **kwargs)
...@@ -8,4 +8,4 @@ class BaseKPICalculatorModel(ABCHasStrictTraits): ...@@ -8,4 +8,4 @@ class BaseKPICalculatorModel(ABCHasStrictTraits):
def __init__(self, bundle, *args, **kwargs): def __init__(self, bundle, *args, **kwargs):
self.bundle = bundle self.bundle = bundle
super(BaseKPICalculatorModel, self).__init__(self, *args, **kwargs) super(BaseKPICalculatorModel, self).__init__(*args, **kwargs)
...@@ -8,4 +8,4 @@ class BaseMCOModel(ABCHasStrictTraits): ...@@ -8,4 +8,4 @@ class BaseMCOModel(ABCHasStrictTraits):
def __init__(self, bundle, *args, **kwargs): def __init__(self, bundle, *args, **kwargs):
self.bundle = bundle self.bundle = bundle
super(BaseMCOModel, self).__init__(self, *args, **kwargs) super(BaseMCOModel, self).__init__(*args, **kwargs)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment