Fix toric code and classical kernels demos

demonstrations/tutorial_classical_kernels.py

@@ -701,7 +701,10 @@ def probabilities_threshold_normalize(probabilities, thresh = 1.e-10):
 # itself, in case it is of use later:
 
 def fourier_q(d, thetas, amplitudes):
-    return np.real(coefficients(lambda x: QK(x, thetas, amplitudes), 1, d-1))
+    def QK_partial(x):
+        squeezed_x = qml.math.squeeze(x)
+        return QK(squeezed_x, thetas, amplitudes)
+    return np.real(coefficients(QK_partial, 1, d-1))
 
 ###############################################################################
 # And with this, we can finally visualize how the Fourier spectrum of the

demonstrations/tutorial_toric_code.py

@@ -448,7 +448,7 @@ def excitations(x_sites, z_sites):
 # expectation value is :math:`-1`, then a quasiparticle exists in that location.
 #
 
-occupation_numbers = lambda expvals: 0.5 * (1 - expvals)
+occupation_numbers = lambda expvals: [0.5 * (1 - val) for val in expvals]
 
 def print_info(x_expvals, z_expvals):
     E = -sum(x_expvals) - sum(z_expvals)