Merge pull request #92 from pasqal-io/qutip-fix

Fixing compatibility issue with latest version of QuTiP
pasqal-io · Feb 26, 2021 · 1dae70b · 1dae70b
2 parents 748a9b8 + 845a12f
commit 1dae70b
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'