diff --git a/src/awkward/_broadcasting.py b/src/awkward/_broadcasting.py index c0c44ab191..a49bc6ff2c 100644 --- a/src/awkward/_broadcasting.py +++ b/src/awkward/_broadcasting.py @@ -1009,6 +1009,7 @@ def continuation(): def broadcast_and_apply( inputs, action, + *, depth_context: dict[str, Any] | None = None, lateral_context: dict[str, Any] | None = None, allow_records: bool = True, diff --git a/src/awkward/operations/ak_concatenate.py b/src/awkward/operations/ak_concatenate.py index d44414c86f..9c4d207893 100644 --- a/src/awkward/operations/ak_concatenate.py +++ b/src/awkward/operations/ak_concatenate.py @@ -296,10 +296,7 @@ def action(inputs, depth, **kwargs): return None out = ak._broadcasting.broadcast_and_apply( - content_or_others, - action, - allow_records=True, - right_broadcast=False, + content_or_others, action, allow_records=True, right_broadcast=False )[0] return wrap_layout(out, behavior, highlevel) diff --git a/src/awkward/operations/ak_mask.py b/src/awkward/operations/ak_mask.py index 697314dea6..1887219f41 100644 --- a/src/awkward/operations/ak_mask.py +++ b/src/awkward/operations/ak_mask.py @@ -122,10 +122,7 @@ def action(inputs, backend, **kwargs): behavior = behavior_of(array, mask, behavior=behavior) out = ak._broadcasting.broadcast_and_apply( - [layoutarray, layoutmask], - action, - numpy_to_regular=True, - right_broadcast=False, + [layoutarray, layoutmask], action, numpy_to_regular=True, right_broadcast=False ) assert isinstance(out, tuple) and len(out) == 1 return wrap_layout(out[0], behavior, highlevel) diff --git a/src/awkward/operations/ak_where.py b/src/awkward/operations/ak_where.py index 1eda08e469..9014dcd8b7 100644 --- a/src/awkward/operations/ak_where.py +++ b/src/awkward/operations/ak_where.py @@ -137,9 +137,7 @@ def action(inputs, **kwargs): return None out = ak._broadcasting.broadcast_and_apply( - [condition_layout, x_layout, y_layout], - action, - numpy_to_regular=True, + [condition_layout, x_layout, y_layout], action, numpy_to_regular=True ) return wrap_layout(out[0], behavior, highlevel) diff --git a/src/awkward/operations/ak_with_field.py b/src/awkward/operations/ak_with_field.py index e9898fa8dd..6193a173a7 100644 --- a/src/awkward/operations/ak_with_field.py +++ b/src/awkward/operations/ak_with_field.py @@ -165,9 +165,7 @@ def action(inputs, **kwargs): return None out = ak._broadcasting.broadcast_and_apply( - [base, what], - action, - right_broadcast=False, + [base, what], action, right_broadcast=False ) assert isinstance(out, tuple) and len(out) == 1 diff --git a/src/awkward/operations/ak_zip.py b/src/awkward/operations/ak_zip.py index b5ade595f3..4dc82efb37 100644 --- a/src/awkward/operations/ak_zip.py +++ b/src/awkward/operations/ak_zip.py @@ -214,7 +214,7 @@ def action(inputs, depth, **ignore): return None out = ak._broadcasting.broadcast_and_apply( - layouts, action, behavior, right_broadcast=right_broadcast + layouts, action, right_broadcast=right_broadcast ) assert isinstance(out, tuple) and len(out) == 1 out = out[0] diff --git a/src/awkward/operations/str/akstr_join.py b/src/awkward/operations/str/akstr_join.py index ff33cbd9fa..49794288c2 100644 --- a/src/awkward/operations/str/akstr_join.py +++ b/src/awkward/operations/str/akstr_join.py @@ -118,8 +118,7 @@ def apply_binary(layouts, **kwargs): ) (out,) = ak._broadcasting.broadcast_and_apply( - (layout, separator_layout), - apply_binary, + (layout, separator_layout), apply_binary ) return wrap_layout(out, highlevel=highlevel, behavior=behavior) diff --git a/src/awkward/operations/str/akstr_repeat.py b/src/awkward/operations/str/akstr_repeat.py index 37d1c3974e..c36dadf627 100644 --- a/src/awkward/operations/str/akstr_repeat.py +++ b/src/awkward/operations/str/akstr_repeat.py @@ -79,7 +79,7 @@ def action(inputs, **kwargs): return (_apply_through_arrow(pc.binary_repeat, *inputs),) (out,) = ak._broadcasting.broadcast_and_apply( - (layout, num_repeats_layout), action, behavior + (layout, num_repeats_layout), action ) else: