From fcd502d05b5b2dbe8781962aeab1e78a8c555148 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Tue, 25 Oct 2022 15:00:21 +0200 Subject: [PATCH] Made ColumbusConfigDefinedExample compatible with humanPlayer.py --- ...igDefinedExample.md => ColumbusConfigDefinedExample.yaml | 6 +++++- columbus/humanPlayer.py | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) rename ColumbusConfigDefinedExample.md => ColumbusConfigDefinedExample.yaml (96%) 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)