From e08accdc22891eb9186eb89bc0bfc61cd40d569f Mon Sep 17 00:00:00 2001 From: zach dwiel Date: Mon, 10 Dec 2018 19:43:57 -0500 Subject: [PATCH] allow case insensitive selected level name matching --- rl_coach/environments/environment.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/rl_coach/environments/environment.py b/rl_coach/environments/environment.py index e29795e..dce3c77 100644 --- a/rl_coach/environments/environment.py +++ b/rl_coach/environments/environment.py @@ -60,10 +60,11 @@ class SingleLevelSelection(LevelSelection): logger.screen.error("No level has been selected. Please select a level using the -lvl command line flag, " "or change the level in the preset. \nThe available levels are: \n{}" .format(', '.join(sorted(self.levels.keys()))), crash=True) - if self.selected_level not in self.levels.keys(): + selected_level = self.selected_level.lower() + if selected_level not in self.levels.keys(): logger.screen.error("The selected level ({}) is not part of the available levels ({})" - .format(self.selected_level, ', '.join(self.levels.keys())), crash=True) - return self.levels[self.selected_level] + .format(selected_level, ', '.join(self.levels.keys())), crash=True) + return self.levels[selected_level] # class SingleLevelPerPhase(LevelSelection):