diff --git a/Palette/env/engine.py b/Palette/env/engine.py index b91634a..f686daf 100644 --- a/Palette/env/engine.py +++ b/Palette/env/engine.py @@ -266,11 +266,11 @@ class SeaEnvEngine(object): self._ship_charging[ship_id] = False self._ship_charging_time_left[ship_id] = 0 self.ship_body[ship_id].angle = 1.5*np.pi - elif action == DOWN: + elif action == DOWN: self._ship_charging[ship_id] = False self._ship_charging_time_left[ship_id] = 0 self.ship_body[ship_id].angle = 0.5*np.pi - elif action == LEFT: # left + elif action == LEFT: self._ship_charging[ship_id] = False self._ship_charging_time_left[ship_id] = 0 self.ship_body[ship_id].angle = np.pi