Skip to content

Commit

Permalink
dak.firsts should be able to take any axis >= 1 in common case
Browse files Browse the repository at this point in the history
  • Loading branch information
lgray authored Nov 30, 2023
1 parent 2dd462b commit 9bde8af
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/dask_awkward/lib/structure.py
Original file line number Diff line number Diff line change
Expand Up @@ -402,7 +402,7 @@ def firsts(
behavior: Mapping | None = None,
attrs: Mapping[str, Any] | None = None,
) -> Any:
if axis == 1:
if axis >= 1:
return map_partitions(
_FirstsFn(axis=axis, highlevel=highlevel, behavior=behavior, attrs=attrs),
array,
Expand Down

0 comments on commit 9bde8af

Please sign in to comment.