steffen/FIKS_Entwicklungsumgebung
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Restructuring

Browse Source
This commit is contained in:
steffen-illium
2021-06-17 16:23:35 +02:00
parent 7b4060a042
commit 7893e1131e
2 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
environments/factory/base_factory.py
View File

@ -210,12 +210,13 @@ class BaseFactory(gym.Env):
# Step the door close intervall
agents_pos = [agent.pos for agent in self._agent_states]
for door_i, door in enumerate(self._door_states):
if door.is_open and door.time_to_close and door.pos not in agents_pos:
door.time_to_close -= 1
elif door.is_open and not door.time_to_close and door.pos not in agents_pos:
door.use()
self._state[self._state_slices.by_name(h.DOORS)] = 1 if door.is_closed else -1
if self.has_doors:
for door_i, door in enumerate(self._door_states):
if door.is_open and door.time_to_close and door.pos not in agents_pos:
door.time_to_close -= 1
elif door.is_open and not door.time_to_close and door.pos not in agents_pos:
door.use()
self._state[self._state_slices.by_name(h.DOORS)] = 1 if door.is_closed else -1
reward, info = self.calculate_reward(self._agent_states)