move flatten to utils

1 files changed, 8 insertions, 0 deletions

diff --git a/lib/utils.py b/lib/utils.py
index 4208c3a..2cf31be 100644
--- a/lib/utils.py
+++ b/lib/utils.py
@@ -21,6 +21,14 @@ def float_or_nan(n):
     except ValueError:
         return np.nan
 
+def flatten(somelist):
+    """
+    Flatten a list.
+
+    Example: flatten([[1, 2], [3], [4, 5]]) -> [1, 2, 3, 4, 5]
+    """
+    return [item for sublist in somelist for item in sublist]
+
 def param_slice_eq(a, b, index):
     """
     Check if by_param keys a and b are identical, ignoring the parameter at index.