add workflow_id as arguments of FCC and EM

Change-Id: I38bef5254828788d6d52458a4a535d0d29bfea67
BlueBrain · May 31, 2024 · 23b509d · 23b509d
1 parent a4db6ac
commit 23b509d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git 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,14 +134,15 @@ 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
         self.passed_validation = passedValidation
         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
@@ -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):