Refactor usability tests.

chenjiaoAngel · chenjiaoAngel · commit 4c116d210fbd · 2025-12-26T20:42:44.000+08:00
Co-authored-by: Shangwei-Li&lt;lishangwei@mail.ustc.edu.cn&gt;
diff --git a/verl/utils/dataset/rl_dataset.py b/verl/utils/dataset/rl_dataset.py
@@ -487,7 +487,8 @@ def split(self, num_splits: int):
         print(f"total_samples: {total_samples}")
         if total_samples == 0 or total_samples % num_splits != 0:
             raise ValueError(
-                f"Cannot split empty dataset or dataset size {total_samples} is not divisible by num_splits {num_splits}"
+                f"Cannot split empty dataset or dataset size {total_samples}"
+                f" is not divisible by num_splits {num_splits}"
             )
 
         split_size = total_samples // num_splits

Original file line number	Diff line number	Diff line change
`@@ -487,7 +487,8 @@ def split(self, num_splits: int):`
`487`	`487`	`print(f"total_samples: {total_samples}")`
`488`	`488`	`if total_samples == 0 or total_samples % num_splits != 0:`
`489`	`489`	`raise ValueError(`
`490`		`- f"Cannot split empty dataset or dataset size {total_samples} is not divisible by num_splits {num_splits}"`
	`490`	`+ f"Cannot split empty dataset or dataset size {total_samples}"`
	`491`	`+ f" is not divisible by num_splits {num_splits}"`
`491`	`492`	`)`
`492`	`493`
`493`	`494`	`split_size = total_samples // num_splits`