From 8f3486e5fc69f2402476b830d21b271a93ed0046 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Mon, 14 Mar 2016 18:25:35 +0000 Subject: Replace lists.collect with lists.flatten --- mammoth/lists.py | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) (limited to 'mammoth/lists.py') 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() -- cgit v1.2.3