fixed multiwalker crashing on completion issue

Farama-Foundation · May 14, 2021 · 121106c · 121106c
1 parent 81125bb
commit 121106c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pettingzoo/sisl/multiwalker/multiwalker_base.py b/pettingzoo/sisl/multiwalker/multiwalker_base.py
@@ -447,7 +447,7 @@ def scroll_subroutine(self):
  rewards += self.terminate_reward
  done = [True] * self.n_walkers
  if pos[0] > (self.terrain_length - TERRAIN_GRASS) * TERRAIN_STEP:
- done = True
+ done = [True] * self.n_walkers
  rewards += self.fall_reward * self.fallen_walkers
  if self.terminate_on_fall and np.sum(self.fallen_walkers) > 0:
  rewards += self.terminate_reward