Specify type for i and check sorting of block rows.

hodgestar · Mar 20, 2023 · da03d45 · da03d45
1 parent 031a093
commit da03d45
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/qutip/core/data/csr.pyx b/qutip/core/data/csr.pyx
@@ -902,15 +902,15 @@ cpdef CSR from_csr_blocks(
 
     cdef base.idxint rowpos, colpos
     cdef base.idxint row_idx, col_idx
-    cdef base.idxint op_i, op_row, op_row_start, op_row_end, op_row_len
+    cdef base.idxint op_i, op_row, op_row_start, op_row_end, op_row_len, i
 
     out.row_index[0] = 0
 
     for row_idx in range(n_blocks):
         prev_op_idx = op_idx
         while op_idx < op_len:
-            #if block_rows[op_idx] < row_idx:
-            #    raise ValueError("Block row indexes (block_rows) are not sorted.")
+            if block_rows[op_idx] < row_idx:
+                raise ValueError("Block row indexes (block_rows) are not sorted.")
             if block_rows[op_idx] != row_idx:
                 break
             op_idx += 1