fix exportJointly option (#35)

RelationalAI · Oct 10, 2023 · 3424c52 · 3424c52
1 parent 93d8d5a
commit 3424c52
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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: