improves: flatten improved (#185)

cereja-project · Nov 18, 2022 · e2d354f · e2d354f
1 parent 37be684
commit e2d354f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cereja/__init__.py b/cereja/__init__.py
@@ -46,7 +46,7 @@
 from . import experimental
 from ._requests import request
 
-VERSION = "1.7.9.final.0"
+VERSION = "1.8.0.final.0"
 
 __version__ = get_version_pep440_compliant(VERSION)
 

diff --git a/cereja/array/_array.py b/cereja/array/_array.py
@@ -307,11 +307,11 @@ def _iter(seq_, take_shapes=False):
             sequence, has_sequence, shape = _iter(sequence, take_shapes=return_shapes)
         except TypeError:
             sequence, has_sequence, shape = _iter(sequence, take_shapes=return_shapes)
-        if not has_sequence:
-            break
         deep += 1
         if return_shapes:
             shapes[deep] = shape
+        if not any(map(is_sequence, sequence)):
+            break
     return (sequence, shapes) if return_shapes else sequence