Merging from main

Author: astroC86

pylops_mpi/basicoperators/BlockDiag.py

@@ -170,8 +170,8 @@ def __init__(self, ops: Sequence[MPILinearOperator], base_comm: MPI.Comm = MPI.C
                  dtype: Optional[DTypeLike] = None):
         self.ops = ops
         dtype = _get_dtype(self.ops) if dtype is None else np.dtype(dtype)
-        shape = (int(np.sum(op.shape[0] for op in ops)),
-                 int(np.sum(op.shape[1] for op in ops)))
+        shape = (int(sum(op.shape[0] for op in ops)),
+                 int(sum(op.shape[1] for op in ops)))
         super().__init__(shape=shape, dtype=dtype, base_comm=base_comm)
 
     def _matvec(self, x: StackedDistributedArray) -> StackedDistributedArray:

pylops_mpi/basicoperators/VStack.py

@@ -183,7 +183,7 @@ def __init__(self, ops: Sequence[MPILinearOperator],
         self.ops = ops
         if len(set(op.shape[1] for op in ops)) > 1:
             raise ValueError("Operators have different number of columns")
-        shape = (int(np.sum(op.shape[0] for op in ops)), ops[0].shape[1])
+        shape = (int(sum(op.shape[0] for op in ops)), ops[0].shape[1])
         dtype = _get_dtype(self.ops) if dtype is None else np.dtype(dtype)
         super().__init__(shape=shape, dtype=dtype, base_comm=base_comm)