Fix bug with thresh=None

CFEL-CMI · Aug 23, 2023 · 9103cd9 · 9103cd9
1 parent b0ad66f
commit 9103cd9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/richmol/rot/labtens.py b/richmol/rot/labtens.py
@@ -155,7 +155,7 @@ def __init__(self, arg, basis, thresh=None, **kwargs):
 
             # self.basis = basis # can be deprecated, replaced by `symtop_basis`
 
-            csr_mat = lambda m, thr: csr_matrix(np.where(np.abs(m) > thr, m, 0))
+            csr_mat = lambda m, thr: csr_matrix(m) if thr==None else csr_matrix(np.where(np.abs(m) > thr, m, 0))
             self.symtop_basis = {
                 round(float(J), 1): {
                     sym: {