samrat-rm commited on
Commit
87f9568
·
1 Parent(s): 3ef6b97

chore: updating the [END] log

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -254,8 +254,8 @@ async def run_episode(
254
 
255
  finally:
256
  steps_taken = len(rewards)
257
- final_reward = f"{rewards[-1]:.2f}" if rewards else "0.10"
258
- print(f"[END] success={str(success).lower()} steps={steps_taken} reward={final_reward}", flush=True)
259
 
260
  return {"scenario_key": scenario_key, "score": score, "steps": steps_taken, "success": success}, env
261
 
 
254
 
255
  finally:
256
  steps_taken = len(rewards)
257
+ rewards_str = ",".join(f"{r:.2f}" for r in rewards) if rewards else "0.10"
258
+ print(f"[END] success={str(success).lower()} steps={steps_taken} rewards={rewards_str}", flush=True)
259
 
260
  return {"scenario_key": scenario_key, "score": score, "steps": steps_taken, "success": success}, env
261