Skip to content

Commit 1ada946

Browse files
committed
fix: training loop example for sumo rl
1 parent 786cd33 commit 1ada946

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/envs/sumo_rl_env/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,7 @@ for episode in range(10):
220220
action_id = np.random.choice(result.observation.action_mask)
221221

222222
# Take action
223-
result = env.step(SumoAction(phase_id=action_id))
223+
result = env.step(SumoAction(phase_id=int(action_id)))
224224

225225
episode_reward += result.reward or 0
226226
steps += 1

0 commit comments

Comments
 (0)