Skip to content

Commit fae15ac

Browse files
committed
Fix type error
1 parent fc09267 commit fae15ac

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

stable_baselines3/her/her_replay_buffer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -402,7 +402,7 @@ def truncate_last_trajectory(self) -> None:
402402
self.dones[self.pos - 1, env_idx] = True
403403
# make sure that last episodes can be sampled and
404404
# update next episode start (self._current_ep_start)
405-
self._compute_episode_length(env_idx)
405+
self._compute_episode_length(int(env_idx))
406406
# handle infinite horizon tasks
407407
if self.handle_timeout_termination:
408408
self.timeouts[self.pos - 1, env_idx] = True # not an actual timeout, but it allows bootstrapping

0 commit comments

Comments
 (0)