Add generatorName in VariantTaskConfig to obey schema (#5)

BlueBrain · Aug 23, 2024 · 46d28c4 · 46d28c4
1 parent 9b315db
commit 46d28c4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/bbp_workflow/generation/entity.py b/bbp_workflow/generation/entity.py
@@ -38,6 +38,7 @@ class GeneratorTaskConfig(BbpWorkflowConfig):
 @attributes(
     {
         "distribution": AttrOf(DataDownload),
+        "generatorName": AttrOf(str, default=None),
     }
 )
 class VariantTaskConfig(BbpWorkflowConfig):

diff --git a/bbp_workflow/generation/generator_base.py b/bbp_workflow/generation/generator_base.py
@@ -188,7 +188,9 @@ def build_variant_task(self, *, algorithm, version, data, output_dir, extra_inpu
         # create or find config that parametrizes the variant definition
         config = create_or_find(
             entity_cls=VariantTaskConfig,
-            properties={},
+            properties={
+                "generatorName": self.generator_name,
+            },
             serialized_content=json.dumps({"inputs": input_values}),
             filename="variant_task_configuration.json",
             content_type="application/json",