[REF] Use divmod

f-dangel · Jul 3, 2024 · eafdb08 · eafdb08
1 parent 84bdc1d
commit eafdb08
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/singd/structures/blockdiagonal.py b/singd/structures/blockdiagonal.py
@@ -291,8 +291,7 @@ def from_mat_inner(cls, X: Tensor) -> BlockDiagonalMatrixTemplate:
             The structured matrix extracted from `X @ X^T`.
         """
         dim = X.shape[0]
-        num_blocks = dim // cls.BLOCK_DIM
-        last_dim = dim - num_blocks * cls.BLOCK_DIM
+        num_blocks, last_dim = divmod(dim, cls.BLOCK_DIM)
         dims = {"block": num_blocks, "row": cls.BLOCK_DIM}
 
         X_blocks, X_last = X.split([num_blocks * cls.BLOCK_DIM, last_dim])