From 94eaaed1975a22a8360c22852b0a25787667e88a Mon Sep 17 00:00:00 2001 From: Tianming Xu Date: Fri, 22 Feb 2019 02:40:28 +0800 Subject: [PATCH] [rllib]convert export format to lower case while validating (#4088) * convert export format to lower case while validating * fix lint error --- python/ray/tune/trial.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/python/ray/tune/trial.py b/python/ray/tune/trial.py index 7f32d3941..9acef4ac1 100644 --- a/python/ray/tune/trial.py +++ b/python/ray/tune/trial.py @@ -217,11 +217,13 @@ class ExportFormat(object): Raises: ValueError if the format is unknown. """ - for export_format in export_formats: - if export_format not in [ + for i in range(len(export_formats)): + export_formats[i] = export_formats[i].strip().lower() + if export_formats[i] not in [ ExportFormat.CHECKPOINT, ExportFormat.MODEL ]: - raise TuneError("Unsupported export format: " + export_format) + raise TuneError("Unsupported export format: " + + export_formats[i]) class Trial(object):