wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-02 00:46:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[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
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/trial.py
+5 -3
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):