extree_data function

Copy the partykit::extree_data function from partykit to avoid dependency issue