`qml.sample` differs in shape when using Catalyst

```python
from catalyst import qjit, measure, cond
import pennylane as qml

@qml.qnode(qml.device("lightning.qubit", wires=1, shots=100))
def circuit():
    return qml.sample()

print(circuit().shape)
print(qjit(circuit)().shape)
```
```bash
$ python test.py 
(100,)
(100, 1)
```
