pramodmisra Claude Opus 4.5 commited on
Commit
76eba39
·
1 Parent(s): a7386cd

Fix done field serialization for OpenEnv compatibility

Browse files

OpenEnv's serialize_observation() expects observation.done but we were
only setting observation.reward. Now we also set done = is_terminal.

Tested: reward=13.50, done=True

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Files changed (1) hide show
  1. server/claims_environment.py +3 -1
server/claims_environment.py CHANGED
@@ -197,8 +197,10 @@ class ClaimsEnvironment(Environment):
197
  self._last_reward = reward
198
  self._state.total_reward += reward
199
 
200
- # Set reward on observation for client serialization
 
201
  observation.reward = reward
 
202
 
203
  return observation
204
 
 
197
  self._last_reward = reward
198
  self._state.total_reward += reward
199
 
200
+ # Set reward and done on observation for OpenEnv serialization
201
+ # OpenEnv's serialize_observation() expects observation.reward and observation.done
202
  observation.reward = reward
203
+ observation.done = observation.is_terminal
204
 
205
  return observation
206