Replaced many maps with for a loop

OpenFreeEnergy · Aug 23, 2024 · c5084de · c5084de
1 parent fe4b87b
commit c5084de
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/alchemiscale/storage/statestore.py b/alchemiscale/storage/statestore.py
@@ -2461,9 +2461,15 @@ def add_protocol_dag_result_ref_traceback(
             except IndexError:
                 raise KeyError("Could not find ProtocolDAGResultRef in database.")
 
-            tracebacks = list(map(lambda puf: puf.traceback, protocol_unit_failures))
-            source_keys = list(map(lambda puf: puf.source_key, protocol_unit_failures))
-            failure_keys = list(map(lambda puf: puf.key, protocol_unit_failures))
+            failure_keys = []
+            source_keys = []
+            tracebacks = []
+
+            for puf in protocol_unit_failures:
+                failure_keys.append(puf.key)
+                source_keys.append(puf.source_key)
+                tracebacks.append(puf.traceback)
+
             traceback = Traceback(tracebacks, source_keys, failure_keys)
 
             _, traceback_node, _ = self._gufe_to_subgraph(