Skip to content

Commit ddc55fd

Browse files
committed
Fix type
1 parent a778daf commit ddc55fd

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

sgkit/utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -362,6 +362,8 @@ def split_array_chunks(n: int, blocks: int) -> Tuple[int, ...]:
362362
if blocks <= 0:
363363
raise ValueError(f"Number of blocks ({blocks}) must be >= 0")
364364
n_div, n_mod = np.divmod(n, blocks)
365+
n_div = int(n_div)
366+
n_mod = int(n_mod)
365367
chunks = n_mod * (n_div + 1,) + (blocks - n_mod) * (n_div,)
366368
return chunks # type: ignore[no-any-return]
367369

0 commit comments

Comments
 (0)