This commit is contained in:
p57492168 2022-07-03 12:36:58 +08:00
parent 6fbd3c65f6
commit 3e24e286f2
1 changed files with 2 additions and 2 deletions

View File

@ -262,11 +262,11 @@ class SeaEnvEngine(object):
raise AssertionError("Cannot continue running after crash!")
for ship_id, action in enumerate(actions):
if action == UP:
if action == UP:
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: # 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