Merge pull request #23 from Ericgig/version_update

Update for latest numpy version

Author: Ericgig

src/qutip_cuquantum/qobjevo.pyx

@@ -91,7 +91,7 @@ cdef class CuQobjEvo(QobjEvo):
             self.expect_ready = True
         # Workaround for a bug in cudensity 0.2.0.
         settings.cuDensity["ctx"].release_workspace()
-        return self.operator.compute_expectation(t, None, state.base).get()
+        return self.operator.compute_expectation(t, None, state.base).get()[0]
 
     def arguments(self, args):
         raise NotImplementedError
@@ -131,3 +131,6 @@ cdef class CuQobjEvo(QobjEvo):
 
     def data(self, t):
         raise NotImplementedError
+
+    def __repr__(self):
+        return "qutip-cuQuantum QobjEvo"

src/qutip_cuquantum/state.py

@@ -245,15 +245,15 @@ def trace_cuState(mat):
     if mat.shape[0] != mat.shape[1]:
         raise ValueError(...)
 
-    return complex(mat.base.trace())
+    return complex(mat.base.trace()[0])
 
 
 @_data.inner.register(CuState)
 def inner_cuState(left, right, scalar_is_ket=False):
     if left.shape == (1, 1) and not scalar_is_ket:
         inner = left.base.storage[0] * right.base.storage[0]
     else:
-        inner = left.base.inner_product(right.base)
+        inner = left.base.inner_product(right.base)[0]
     return complex(inner)
 
 
@@ -288,7 +288,7 @@ def iadd_cuState(left, right, scale=1.):
 
 @_data.norm.frobenius.register(CuState)
 def frobenius_cuState(mat):
-    return float(mat.base.norm())**0.5
+    return float(mat.base.norm()[0])**0.5
 
 
 @_data.norm.l2.register(CuState)
@@ -307,7 +307,7 @@ def wrmn_error_cuState(diff, state, atol, rtol):
         )
     diff.base.storage[:] = cp.abs(diff.base.storage)
     diff.base.storage[:] = diff.base.storage / (atol + rtol * cp.abs(state.base.storage))
-    return float(diff.base.norm() / (diff.shape[0] * diff.shape[1]))**0.5
+    return float(diff.base.norm()[0] / (diff.shape[0] * diff.shape[1]))**0.5
 
 
 @_data.reshape.register(CuState)