[Bug] The dynamic one-shot transform does not preserve pytree output of QNodes

[josh146]

dev = qml.device("lightning.qubit", wires=1, shots=20)

@qml.qjit
@qml.qnode(dev, mcm_method="one-shot")
def func(x):
    qml.RX(x, wires=0)
    m_0 = catalyst.measure(0, postselect=1)
    return {"hi": qml.expval(qml.Z(0))}

>>> func(0.9)
array(1.)