Skip to content

Commit

Permalink
[df] Correctly clone AsNumpyResult in distributed execution
Browse files Browse the repository at this point in the history
  • Loading branch information
vepadulano committed Mar 28, 2024
1 parent 1edca86 commit 62a7ea3
Showing 1 changed file with 2 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from typing import Iterable, Set, Tuple

import ROOT
from ROOT._pythonization._rdataframe import AsNumpyResult
from ROOT._pythonization._rdataframe import AsNumpyResult, _clone_asnumpyresult

from DistRDF.PythonMergeables import SnapshotResult

Expand Down Expand Up @@ -254,11 +254,7 @@ def clone_action(result_promise, _):

@clone_action.register(AsNumpyResult)
def _(asnumpyres, _):
asnumpyres._result_ptrs = {
col: ROOT.Internal.RDF.CloneResultAndAction(ptr)
for (col, ptr) in asnumpyres._result_ptrs.items()
}
return asnumpyres
return _clone_asnumpyresult(asnumpyres)


@clone_action.register(SnapshotResult)
Expand Down

0 comments on commit 62a7ea3

Please sign in to comment.