We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0f35746 commit 3302cd9Copy full SHA for 3302cd9
src/llmcompressor/datasets/utils.py
@@ -403,9 +403,9 @@ def get_rank_partition(split: str, num_samples: int) -> str:
403
we give each device at least S//D samples and distribute
404
the remaining samples as evenly as possible across all devices
405
"""
406
- assert (
407
- "[" not in split
408
- ), "Split string should not already contain partitioning brackets"
+ assert "[" not in split, (
+ "Split string should not already contain partitioning brackets"
+ )
409
410
start, end = _get_partition_start_end(
411
num_samples, dist.get_rank(), dist.get_world_size()
0 commit comments