Skip to content

Commit cebcc3e

Browse files
authored
Merge pull request #236 from lincc-frameworks/fix-flat_length
Fix NestedExtensionArray.flat_length for empty arrays
2 parents b59d701 + 743d62c commit cebcc3e

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

src/nested_pandas/series/ext_array.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -798,7 +798,10 @@ def list_lengths(self) -> np.ndarray:
798798
@property
799799
def flat_length(self) -> int:
800800
"""Length of the flat arrays"""
801-
return pa.compute.sum(self.list_lengths).as_py()
801+
sum_result = pa.compute.sum(self.list_lengths).as_py()
802+
if sum_result is None:
803+
sum_result = 0
804+
return sum_result
802805

803806
@property
804807
def num_chunks(self) -> int:

tests/nested_pandas/nestedframe/test_nestedframe.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1277,3 +1277,9 @@ def test_issue193():
12771277
ndf.query("nested.flux / nested.t > 0")
12781278
# This failed with numpy 1 with:
12791279
# TypeError: Cannot interpret 'double[pyarrow]' as a data type
1280+
1281+
1282+
def test_issue235():
1283+
"""https://github.com/lincc-frameworks/nested-pandas/issues/235"""
1284+
nf = generate_data(3, 10).iloc[:0]
1285+
nf["nested.x"] = []

0 commit comments

Comments
 (0)