diff --git a/src/aspire/operators/blk_diag_matrix.py b/src/aspire/operators/blk_diag_matrix.py index 7612d38c38..3e493f9525 100644 --- a/src/aspire/operators/blk_diag_matrix.py +++ b/src/aspire/operators/blk_diag_matrix.py @@ -91,7 +91,7 @@ def append(self, blk): :param blk: Block to append (ndarray). """ - self.data.append(blk) + self.data.append(blk.astype(self.dtype, copy=False)) self.nblocks += 1 self.reset_cache() @@ -130,7 +130,7 @@ def __setitem__(self, key, value): Convenience wrapper, setter on self.data. """ - self.data[key] = value + self.data[key] = value.astype(self.dtype, copy=False) self.reset_cache() def __len__(self):