steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-12-12 10:30:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

major redesign ob observations and entittes

Browse Source
This commit is contained in:
Steffen Illium
2023-06-09 14:04:17 +02:00
parent 901fbcbc32
commit c552c35f66
161 changed files with 4458 additions and 4163 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
reload_agent.py
View File

@@ -2,13 +2,13 @@ import warnings
from pathlib import Path
import yaml
from stable_baselines3 import A2C, PPO, DQN
from stable_baselines3 import PPO
from environments.factory.additional.dirt.dirt_util import Constants
from environment.factory import BaseFactory
from environment.logging.envmonitor import EnvMonitor
from environment.logging.recorder import EnvRecorder
from environments.factory.additional.dirt.factory_dirt import DirtFactory
from environments.logging.envmonitor import EnvMonitor
from environments.logging.recorder import EnvRecorder
from modules.doors import constants as d
warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning)
@@ -42,7 +42,7 @@ if __name__ == '__main__':
pass
# Init Env
with DirtFactory(**env_kwargs) as env:
with BaseFactory(**env_kwargs) as env:
env = EnvMonitor(env)
env = EnvRecorder(env) if record else env
obs_shape = env.observation_space.shape
@@ -62,7 +62,7 @@ if __name__ == '__main__':
if render:
env.render()
try:
door = next(x for x in env.unwrapped.unwrapped[Constants.DOORS] if x.is_open)
door = next(x for x in env.unwrapped.unwrapped[d.DOORS] if x.is_open)
print('openDoor found')
except StopIteration:
pass