Skip to content

Commit

Permalink
fix: re-evaluate!
Browse files Browse the repository at this point in the history
  • Loading branch information
agoose77 committed Oct 4, 2023
1 parent 7d193ff commit c680b1c
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion src/awkward/_connect/numexpr.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,8 +141,13 @@ def action(inputs, **ignore):
or not isinstance(x, ak.contents.Content)
for x in inputs
):
input_primitives = [
x.data if isinstance(x, ak.contents.NumpyArray) else x for x in inputs
]
return (
ak.contents.NumpyArray(numexpr.re_evaluate(dict(zip(names, inputs)))),
ak.contents.NumpyArray(
numexpr.re_evaluate(dict(zip(names, input_primitives)))
),
)
else:
return None
Expand Down

0 comments on commit c680b1c

Please sign in to comment.