diff --git a/bluepyemodel/emodel_pipeline/emodel.py b/bluepyemodel/emodel_pipeline/emodel.py index 57c349d9..2be9e971 100644 --- a/bluepyemodel/emodel_pipeline/emodel.py +++ b/bluepyemodel/emodel_pipeline/emodel.py @@ -120,6 +120,7 @@ def __init__( passedValidation=None, seed=None, emodel_metadata=None, + workflow_id=None, ): """Init @@ -133,6 +134,7 @@ def __init__( did it pass it successfully (None: no validation, True: passed, False: didn't pass) seed (int): seed used during optimisation for this emodel. emodel_metadata (EModelMetadata): metadata of the model (emodel name, etype, ttype, ...) + workflow_id (str): EModelWorkflow id on nexus. """ self.emodel_metadata = emodel_metadata @@ -140,7 +142,7 @@ def __init__( self.fitness = fitness self.seed = seed - self.workflow_id = None + self.workflow_id = workflow_id if isinstance(parameter, dict): self.parameters = parameter diff --git a/bluepyemodel/evaluation/fitness_calculator_configuration.py b/bluepyemodel/evaluation/fitness_calculator_configuration.py index 3159bca3..b57f247d 100644 --- a/bluepyemodel/evaluation/fitness_calculator_configuration.py +++ b/bluepyemodel/evaluation/fitness_calculator_configuration.py @@ -91,6 +91,7 @@ def __init__( validation_protocols=None, stochasticity=False, ion_variables=None, + workflow_id=None, ): """Init. @@ -175,6 +176,7 @@ def __init__( for these protocols, and deterministic for the other ones. ion_variables (list of str): ion current names and ionic concentration anmes for all available mechanisms + workflow_id (str): EModelWorkflow id on nexus. """ self.rmp_duration = 500.0 @@ -209,7 +211,7 @@ def __init__( self.name_rmp_protocol = name_rmp_protocol self.name_rin_protocol = name_rin_protocol - self.workflow_id = None + self.workflow_id = workflow_id self.default_std_value = default_std_value def initialise_protocols(self, protocols):