diff --git a/cli-e2e-test/config/model/scenario1.json b/cli-e2e-test/config/model/scenario1.json index ec4c4ae..79bf3d8 100644 --- a/cli-e2e-test/config/model/scenario1.json +++ b/cli-e2e-test/config/model/scenario1.json @@ -82,7 +82,7 @@ { "type": "Export", "name": "Export", - "exportJointly": false, + "exportJointly": true, "dateFormat": "%Y%m%d", "defaultContainer": "export", "exports": [ diff --git a/workflow/executor.py b/workflow/executor.py index 798b15f..40456f4 100644 --- a/workflow/executor.py +++ b/workflow/executor.py @@ -395,9 +395,10 @@ def execute(self, logger: logging.Logger, env_config: EnvConfig, rai_config: Rai exports = list(filter(lambda e: self._should_export(logger, rai_config, e), self.exports)) if self.export_jointly: exports.sort(key=lambda e: e.container.name) - container_groups = {container: list(group) for container, group in + container_groups = {container_name: list(group) for container_name, group in groupby(exports, key=lambda e: e.container.name)} - for container, grouped_exports in container_groups.items(): + for container_name, grouped_exports in container_groups.items(): + container = env_config.get_container(container_name) ExportWorkflowStep.get_export_function(container)(logger, rai_config, grouped_exports, self.end_date, self.date_format, container) else: