diff --git a/tests/test_jl.py b/tests/test_jl.py index 9251f83..0939427 100644 --- a/tests/test_jl.py +++ b/tests/test_jl.py @@ -12,7 +12,7 @@ def get_max_chunk_size( batch_size: int, -) -> tuple[int, list]: +): max_chunk_size = np.iinfo(np.uint32).max // batch_size return max_chunk_size