diff --git a/bbp_workflow/generation/entity.py b/bbp_workflow/generation/entity.py index 68794a9..1369565 100644 --- 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 index 3b12649..5a848f9 100644 --- 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",