Port (#111)

* fix: bug with map iterate on * fix: Porting to 3.7 * fix: Porting to 3.7 * fix: test compatibility between 3.7 and 3.8
AstraZeneca · Jun 5, 2023 · 38ccf63 · 38ccf63
1 parent a7133fe
commit 38ccf63
Show file tree

Hide file tree

Showing 5 changed files with 525 additions and 557 deletions.
diff --git a/magnus/nodes.py b/magnus/nodes.py
@@ -975,13 +975,14 @@ def execute_as_graph(self, executor, map_variable: dict = None, **kwargs):
             map_variable (dict): The map variables the graph belongs to
             **kwargs: Optional kwargs passed around
         """
-        run_log = executor.run_log_store.get_run_log_by_id(executor.run_id)
-        if self.iterate_on not in run_log.parameters:
+        iterate_on = None
+        try:
+            iterate_on = executor.run_log_store.get_parameters(executor.run_id)[self.iterate_on]
+        except KeyError:
             raise Exception(
                 f"Expected parameter {self.iterate_on} not present in Run Log parameters, was it ever set before?"
             )
 
-        iterate_on = run_log.parameters[self.iterate_on]
         if not isinstance(iterate_on, list):
             raise Exception("Only list is allowed as a valid iterator type")