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

moved renderer.py to base, added initial salina experiments

Browse Source
This commit is contained in:
romue
2021-11-12 13:47:53 +01:00
parent f625b9d8a5
commit b6bda84033
7 changed files with 105 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
environments/factory/__init__.py
View File

@ -1,4 +1,4 @@
def make(env_str, n_agents=1, pomdp_r=2, max_steps=400):
def make(env_str, n_agents=1, pomdp_r=2, max_steps=400, stack_n_frames=3):
import yaml
from pathlib import Path
from environments.factory.combined_factories import DirtItemFactory
@ -9,7 +9,8 @@ def make(env_str, n_agents=1, pomdp_r=2, max_steps=400):
with (Path(__file__).parent / 'levels' / 'parameters' / f'{env_str}.yaml').open('r') as stream:
dictionary = yaml.load(stream, Loader=yaml.FullLoader)
obs_props = ObservationProperties(render_agents=AgentRenderOptions.COMBINED, frames_to_stack=0, pomdp_r=pomdp_r)
obs_props = ObservationProperties(render_agents=AgentRenderOptions.COMBINED,
frames_to_stack=stack_n_frames, pomdp_r=pomdp_r)
factory_kwargs = dict(n_agents=n_agents, max_steps=max_steps, obs_prop=obs_props,
mv_prop=MovementProperties(**dictionary['movement_props']),
@ -17,4 +18,4 @@ def make(env_str, n_agents=1, pomdp_r=2, max_steps=400):
record_episodes=False, verbose=False, **dictionary['factory_props']
)
return DirtFactory(**factory_kwargs)
return DirtFactory(**factory_kwargs).__enter__()