Remove redundant np.array conversion

dbobrenko · dbobrenko · commit e511d0fa65fc · 2025-04-26T00:04:43.000Z
diff --git a/prompting/rewards/exact_match.py b/prompting/rewards/exact_match.py
@@ -175,7 +175,7 @@ async def reward(
         logger.info(f"Success responses: {len(rewards[rewards > 0])}/{len(rewards)}")
 
         reward_output = BatchRewardOutput(
-            rewards=np.array(rewards),
+            rewards=rewards,
             timings=np.array(timing_outputs),
         )
         logger.debug(f"Logits rewards: {reward_output.model_dump()}")

Original file line number	Diff line number	Diff line change
`@@ -175,7 +175,7 @@ async def reward(`
`175`	`175`	`logger.info(f"Success responses: {len(rewards[rewards > 0])}/{len(rewards)}")`
`176`	`176`
`177`	`177`	`reward_output = BatchRewardOutput(`
`178`		`- rewards=np.array(rewards),`
	`178`	`+ rewards=rewards,`
`179`	`179`	`timings=np.array(timing_outputs),`
`180`	`180`	`)`
`181`	`181`	`logger.debug(f"Logits rewards: {reward_output.model_dump()}")`