Spaces:
Sleeping
Sleeping
Commit ·
76eba39
1
Parent(s): a7386cd
Fix done field serialization for OpenEnv compatibility
Browse filesOpenEnv'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>
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
|
|
|
|
| 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 |
|