From 845a12f6ee7393681fb43c5ae379227c19c0f94d Mon Sep 17 00:00:00 2001
From: HSilveri <henrique.silverio@tecnico.ulisboa.pt>
Date: Fri, 26 Feb 2021 12:02:11 +0100
Subject: [PATCH] Fixing compatibility issue with latest version of qutip

---
 pulser/simresults.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pulser/simresults.py b/pulser/simresults.py
index 0c5df7d4..aa7f521b 100644
--- 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'