Change output to QuantumState class for quimb backend

This commit is contained in:
vinitha-balachandran
2024-02-22 18:11:48 +08:00
parent 5f053b25e1
commit 151a5a51a1

View File

@@ -1,6 +1,6 @@
from qibo.backends.numpy import NumpyBackend from qibo.backends.numpy import NumpyBackend
from qibo.config import raise_error from qibo.config import raise_error
from qibo.result import CircuitResult from qibo.result import QuantumState
class QuimbBackend(NumpyBackend): class QuimbBackend(NumpyBackend):
@@ -80,5 +80,4 @@ class QuimbBackend(NumpyBackend):
if return_array: if return_array:
return state.flatten() return state.flatten()
else: else:
circuit._final_state = CircuitResult(self, circuit, state.flatten(), nshots) return QuantumState(state.flatten())
return circuit._final_state