Skip to content

Commit 3302cd9

Browse files
committed
Run make style to fix formatting
Auto-formatting changes from ruff Signed-off-by: rtj1 <tharunjagarlamudi@gmail.com>
1 parent 0f35746 commit 3302cd9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/llmcompressor/datasets/utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -403,9 +403,9 @@ def get_rank_partition(split: str, num_samples: int) -> str:
403403
we give each device at least S//D samples and distribute
404404
the remaining samples as evenly as possible across all devices
405405
"""
406-
assert (
407-
"[" not in split
408-
), "Split string should not already contain partitioning brackets"
406+
assert "[" not in split, (
407+
"Split string should not already contain partitioning brackets"
408+
)
409409

410410
start, end = _get_partition_start_end(
411411
num_samples, dist.get_rank(), dist.get_world_size()

0 commit comments

Comments
 (0)