Merge pull request qutip#2557 from eendebakpt/numpy_version_check

Improve performance of qt.Qobj by static version check
gsuarezr · Nov 19, 2024 · edac7d9 · edac7d9
2 parents b44b697 + 13ef614
commit edac7d9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/doc/changes/2557.misc b/doc/changes/2557.misc
@@ -0,0 +1 @@
+Improve performance of qt.Qobj by using static numpy version check 
diff --git a/qutip/core/data/dense.pyx b/qutip/core/data/dense.pyx
@@ -37,10 +37,11 @@ __all__ = [
 class OrderEfficiencyWarning(EfficiencyWarning):
     pass
 
+is_numpy1 = np.lib.NumpyVersion(np.__version__) < '2.0.0b1'
 
 cdef class Dense(base.Data):
     def __init__(self, data, shape=None, copy=True):
-        if np.lib.NumpyVersion(np.__version__) < '2.0.0b1':
+        if is_numpy1:
             # np2 accept None which act as np1's False
             copy = builtins.bool(copy)
         base = np.array(data, dtype=np.complex128, order='K', copy=copy)
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Improve performance of qt.Qobj by using static numpy version check