Replace lists.collect with lists.flatten

1 files changed, 2 insertions, 11 deletions

diff --git a/mammoth/lists.py b/mammoth/lists.py
index e8477cb..db1f9ca 100644
--- a/mammoth/lists.py
+++ b/mammoth/lists.py
@@ -1,19 +1,10 @@
 import sys
 
 
-def collect(values):
-    result = []
-    _collect(values, result)
-    return result
+def flatten(values):
+    return flat_map(lambda x: x, values)
 
 
-def _collect(values, result):
-    for value in values:
-        if isinstance(value, list):
-            _collect(value, result)
-        elif value is not None:
-            result.append(value)
-
 def unique(values):
     output = []
     seen = set()