diff --git a/force_bdss/api.py b/force_bdss/api.py index 966c2906a13ca6f2fcf8428577620262a83605c5..f9afd7c2e5c4f9c4bb06348a9c3834994c71c760 100644 --- a/force_bdss/api.py +++ b/force_bdss/api.py @@ -1,5 +1,5 @@ from .base_extension_plugin import BaseExtensionPlugin # noqa -from .ids import factory_id, plugin_id # noqa +from .ids import factory_id, plugin_id, mco_parameter_id # noqa from .core.data_value import DataValue # noqa from .core.workflow import Workflow # noqa diff --git a/force_bdss/mco/base_mco_communicator.py b/force_bdss/mco/base_mco_communicator.py index 0f53bec19ccfd5f62b051265c6bedfc6c2f455f3..d946e520d8a6c19adc2163d42028463642a22e9d 100644 --- a/force_bdss/mco/base_mco_communicator.py +++ b/force_bdss/mco/base_mco_communicator.py @@ -21,6 +21,7 @@ class BaseMCOCommunicator(ABCHasStrictTraits): def __init__(self, factory, *args, **kwargs): self.factory = factory + super(BaseMCOCommunicator, self).__init__(*args, **kwargs) super(BaseMCOCommunicator, self).__init__(*args, **kwargs) diff --git a/force_bdss/mco/parameters/base_mco_parameter_factory.py b/force_bdss/mco/parameters/base_mco_parameter_factory.py index ecce688715c8ce2fdcb0764bf0dabae2f05c215d..6ea3ceba0700860028b65a2a17b9b6065dbb228c 100644 --- a/force_bdss/mco/parameters/base_mco_parameter_factory.py +++ b/force_bdss/mco/parameters/base_mco_parameter_factory.py @@ -27,9 +27,9 @@ class BaseMCOParameterFactory(HasStrictTraits): # The model class to instantiate when create_model is called. model_class = Type('BaseMCOParameter') - def __init__(self, mco_factory): + def __init__(self, mco_factory, *args, **kwargs): self.mco_factory = mco_factory - super(BaseMCOParameterFactory, self).__init__() + super(BaseMCOParameterFactory, self).__init__(*args, **kwargs) def create_model(self, data_values=None): """Creates the instance of the model class and returns it.