Fixing compatibility issue with latest version of qutip

pasqal-io · Feb 26, 2021 · 845a12f · 845a12f
1 parent 748a9b8
commit 845a12f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pulser/simresults.py b/pulser/simresults.py
@@ -125,7 +125,7 @@ def sample_final_state(self, meas_basis=None, N_samples=1000):
 
         N = self._size
         self.N_samples = N_samples
-        probs = np.abs(self._states[-1])**2
+        probs = np.abs(self._states[-1].data.toarray())**2
         if self._dim == 2:
             if meas_basis == self._basis_name:
                 # State vector ordered with r first for 'ground_rydberg'