def tensorize(tree_batch): set_batch_nodeID(tree_batch) # we don't want to change itx, itx related to peak information jtenc_holder, mess_dict = Encoder.tensorize(tree_batch) jtenc_holder = jtenc_holder return tree_batch, jtenc_holder