Not possible to do many measurement at the same time anymore?

PennyLane Help
1

Hi,
Today, my code does not work anymore. It seems that it is no longer possible to measure several observables at the same time.
Here is my code:

@jax.jit
def Circuits_Observable_listA(params, inputs):
    dev = qml.device('default.qubit.jax', wires=n_qubits//2)  
    @jax.jit  
    @qml.qnode(dev, interface='jax', diff_method="backprop")
    def qnode(params, inputs):
        for i in range(n_qubits//2):
          qml.RX(jnp.pi*inputs[i], wires=i)
        brick_wall_entangling(params)
        return [qml.expval(Obs) for Obs in H_overlap_A]
    return qnode(params, inputs)

and I get the error:

AttributeError                            Traceback (most recent call last)
/usr/local/lib/python3.8/dist-packages/autoray/autoray.py in shape(x)
    879 @compose
    880 def shape(x):
--> 881     return x.shape
    882 
    883 

AttributeError: 'range' object has no attribute 'shape'

Is there any change in the library?
Thanks a lot

2

Hi @paulin_ds, what version of PennyLane are you using? You should be using version 0.29.1 which is the latest version. You can upgrade the version by running python -m pip install -U pennylane

With that version, the following code should run properly

n_qubits = 2
dev = qml.device('default.qubit.jax', wires=n_qubits)  

@jax.jit
@qml.qnode(dev, interface='jax', diff_method="backprop")
def my_quantum_function(x, y):
    qml.RZ(x, wires=0)
    qml.CNOT(wires=[0, 1])
    qml.RY(y, wires=1)
    return [qml.expval(qml.PauliZ(i)) for i in range(2)]

my_quantum_function(1, 2)

If the version is not the issue please feel free to post a minimal but self-contained version of your code so that I can try to replicate your problem.