hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 18:41:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[rllib]convert export format to lower case while validating (#4088)

Browse source 
* convert export format to lower case while validating

* fix lint error
This commit is contained in:
Tianming Xu 2019-02-22 02:40:28 +08:00 • committed by Eric Liang
parent 2e30f7ba38
commit 94eaaed197
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
python/ray/tune/trial.py
View file

@ -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):