Skip to content

Commit bc244a8

Browse files
nv-alichengclaude
andcommitted
fix: correct f-string upper bound in test_data_loader assertion
The error message said `* 0.2` but the assertion checks `* 1.2`. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 0975b14 commit bc244a8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unit/dataset_manager/test_data_loader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def test_random_data_loader(range_ratio):
8585
assert (
8686
len(sample["input_tokens"]) > input_seq_length * range_ratio * 0.8
8787
and len(sample["input_tokens"]) <= input_seq_length * 1.2
88-
), f"Expected {input_seq_length * range_ratio * 0.8} to {input_seq_length * 0.2} input tokens, got {len(sample['input_tokens'])}"
88+
), f"Expected {input_seq_length * range_ratio * 0.8} to {input_seq_length * 1.2} input tokens, got {len(sample['input_tokens'])}"
8989

9090
assert (
9191
len(sample["prompt"]) >= 1024 * range_ratio * 0.5

0 commit comments

Comments
 (0)