chore: move length check up

ibis/backends/dask/execution/arrays.py

@@ -38,7 +38,11 @@
 @execute_node.register(ops.Array, tuple)
 def execute_array_column(op, cols, **kwargs):
     vals = [execute(arg, **kwargs) for arg in cols]
+
     length = next((len(v) for v in vals if isinstance(v, dd.Series)), None)
+    if length is None:
+        return vals
+
     n_partitions = next((v.npartitions for v in vals if isinstance(v, dd.Series)), None)
 
     def ensure_series(v):
@@ -47,8 +51,6 @@ def ensure_series(v):
         else:
             return dd.from_pandas(pd.Series([v] * length), npartitions=n_partitions)
 
-    if length is None:
-        return vals
     # dd.concat() can only handle array-likes.
     # If we're given a scalar, we need to broadcast it as a Series.
     df = dd.concat([ensure_series(v) for v in vals], axis=1)

ibis/backends/pandas/execution/arrays.py

@@ -22,14 +22,15 @@ def execute_array(op, cols, **kwargs):
     vals = [execute(arg, **kwargs) for arg in cols]
     length = next((len(v) for v in vals if isinstance(v, pd.Series)), None)
 
+    if length is None:
+        return vals
+
     def ensure_series(v):
         if isinstance(v, pd.Series):
             return v
         else:
             return pd.Series(v, index=range(length))
 
-    if length is None:
-        return vals
     # pd.concat() can only handle array-likes.
     # If we're given a scalar, we need to broadcast it as a Series.
     df = pd.concat([ensure_series(v) for v in vals], axis=1)