simplified outputDict.items()

neurolib-dev · Mar 1, 2024 · c75b875 · c75b875
1 parent 69c5c30
commit c75b875
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/neurolib/models/model.py b/neurolib/models/model.py
@@ -593,11 +593,9 @@ def xr(self, group=""):
         assert len(timeDictKey) > 0, f"No time array found (starting with t) in output group {group}."
         t = outputDict[timeDictKey].copy()
         del outputDict[timeDictKey]
-        outputs = []
-        outputNames = []
-        for key, value in outputDict.items():
-            outputNames.append(key)
-            outputs.append(value)
+
+        outputNames, outputs = zip(*outputDict.items())
+        outputNames = list(outputNames)
 
         nNodes = outputs[0].shape[0]
         nodes = list(range(nNodes))