Made ColumbusConfigDefinedExample compatible with humanPlayer.py

This commit is contained in:
Dominik Moritz Roth 2022-10-25 15:00:21 +02:00
parent df8b2d836e
commit fcd502d05b
2 changed files with 6 additions and 2 deletions

View File

@ -1,4 +1,7 @@
env_args: name: Example
params:
task:
env_args:
observable: observable:
- type: State - type: State
coordsAgent: True coordsAgent: True
@ -43,3 +46,4 @@
aux_reward_max: 1 aux_reward_max: 1
aux_penalty_max: 0.1 aux_penalty_max: 0.1
void_damage: 5 void_damage: 5
---

View File

@ -28,7 +28,7 @@ def loadConfigDefinedEnv(EnvClass):
p = input('[Path to config> ') p = input('[Path to config> ')
with open(p, 'r') as f: with open(p, 'r') as f:
docs = list([d for d in yaml.safe_load_all( docs = list([d for d in yaml.safe_load_all(
f) if 'name' in d and d['name'] not in ['SLURM']]) f) if d and 'name' in d and d['name'] not in ['SLURM']])
for i, doc in enumerate(docs): for i, doc in enumerate(docs):
name = doc['name'] name = doc['name']
print('['+str(i)+'] '+name) print('['+str(i)+'] '+name)