Merge pull request #72 from rshin/master

Improve error message for missing problems
tensorflow · Jun 30, 2017 · f95b7c9 · f95b7c9
2 parents 7087807 + 1c775fa
commit f95b7c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensor2tensor/data_generators/problem_hparams.py b/tensor2tensor/data_generators/problem_hparams.py
@@ -78,7 +78,7 @@ def parse_problem_name(problem_name):
 
 def _lookup_problem_hparams_fn(name):
   if name not in PROBLEM_HPARAMS_MAP:
-    map_str = "\n* ".join(PROBLEM_HPARAMS_MAP.keys())
+    map_str = "* " + "\n* ".join(sorted(PROBLEM_HPARAMS_MAP.keys()))
     error_msg = "%s not in the supported set of problems:\n%s" % (name, map_str)
     raise ValueError(error_msg)
   return PROBLEM_HPARAMS_MAP.get(name)