diff --git a/ColumbusConfigDefinedExample.md b/ColumbusConfigDefinedExample.yaml similarity index 96% rename from ColumbusConfigDefinedExample.md rename to ColumbusConfigDefinedExample.yaml index e165024..dd74b75 100644 --- a/ColumbusConfigDefinedExample.md +++ b/ColumbusConfigDefinedExample.yaml @@ -1,4 +1,7 @@ - env_args: +name: Example +params: + task: + env_args: observable: - type: State coordsAgent: True @@ -43,3 +46,4 @@ aux_reward_max: 1 aux_penalty_max: 0.1 void_damage: 5 +--- diff --git a/columbus/humanPlayer.py b/columbus/humanPlayer.py index 4c9095f..d2d2519 100644 --- a/columbus/humanPlayer.py +++ b/columbus/humanPlayer.py @@ -28,7 +28,7 @@ def loadConfigDefinedEnv(EnvClass): p = input('[Path to config> ') with open(p, 'r') as f: 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): name = doc['name'] print('['+str(i)+'] '+name)