Deal with garbage in a, Deal with tuples

KECB · Apr 17, 2017 · 1a53f6f · 1a53f6f
1 parent d0a0ce6
commit 1a53f6f
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/array/flatten.py b/array/flatten.py
@@ -15,13 +15,9 @@
 
 
 def list_flatten(l, a=None):
-    # check a
-    if a is None:
-        # initialize with empty list
-        a = []
-
+    a = list(a) if isinstance(a, (list, tuple)) else []
     for i in l:
-        if isinstance(i, list):
+        if isinstance(i, (list, tuple)):
             list_flatten(i, a)
         else:
             a.append(i)