From 82f91c44d67d960618e1bdfd56e55ac6b93b44f3 Mon Sep 17 00:00:00 2001 From: Mathew Han Date: Wed, 15 Apr 2026 14:22:11 -0700 Subject: [PATCH] [fix] select sample from list for logging --- slime/rollout/sglang_rollout.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/slime/rollout/sglang_rollout.py b/slime/rollout/sglang_rollout.py index b2a3de5678..44c03858ff 100644 --- a/slime/rollout/sglang_rollout.py +++ b/slime/rollout/sglang_rollout.py @@ -573,10 +573,11 @@ async def eval_rollout_single_dataset( for coro in asyncio.as_completed(tasks): sample = await coro if do_print: + logged_sample = sample[0] if isinstance(sample, list) else sample logger.info( "eval_rollout_single_dataset example data: " - f"{[str(sample.prompt) + sample.response]} " - f"reward={sample.reward}" + f"{[str(logged_sample.prompt) + logged_sample.response]} " + f"reward={logged_sample.reward}" ) do_print = False if isinstance(sample, list):