From 4e9c7e9f8a995e116e7afaa02789bd8d64ca16d5 Mon Sep 17 00:00:00 2001 From: Iason Krommydas Date: Sat, 23 Aug 2025 03:47:43 +0300 Subject: [PATCH 1/2] Remove known scalar optimization from dak.num --- src/dask_awkward/lib/structure.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/dask_awkward/lib/structure.py b/src/dask_awkward/lib/structure.py index 0c598fc8..63cb5170 100644 --- a/src/dask_awkward/lib/structure.py +++ b/src/dask_awkward/lib/structure.py @@ -635,9 +635,6 @@ def num( if not highlevel: raise ValueError("Only highlevel=True is supported") if axis == 0 or axis == -1 * array.ndim: - if array.known_divisions: - return new_known_scalar(array.defined_divisions[-1], label="num") - per_axis = map_partitions( ak.num, array, From 5009e45dff33e7463bf3890f587b549679d90ed5 Mon Sep 17 00:00:00 2001 From: Iason Krommydas Date: Sat, 23 Aug 2025 03:48:51 +0300 Subject: [PATCH 2/2] We don't need this here --- src/dask_awkward/lib/structure.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/dask_awkward/lib/structure.py b/src/dask_awkward/lib/structure.py index 63cb5170..c15019ef 100644 --- a/src/dask_awkward/lib/structure.py +++ b/src/dask_awkward/lib/structure.py @@ -20,7 +20,6 @@ PartitionCompatibility, _map_partitions, map_partitions, - new_known_scalar, new_scalar_object, partition_compatibility, )