4 new dialog problems

tensor2tensor/data_generators/all_problems.py

@@ -19,7 +19,6 @@
 from __future__ import print_function
 
 import importlib
-import six
 from six.moves import range # pylint: disable=redefined-builtin
 
 MODULES = [
@@ -39,6 +38,10 @@
  "tensor2tensor.data_generators.cola",
  "tensor2tensor.data_generators.common_voice",
  "tensor2tensor.data_generators.desc2code",
+ "tensor2tensor.data_generators.dialog_cornell",
+ "tensor2tensor.data_generators.dialog_dailydialog",
+ "tensor2tensor.data_generators.dialog_opensubtitles",
+ "tensor2tensor.data_generators.dialog_personachat",
  "tensor2tensor.data_generators.enwik8",
  "tensor2tensor.data_generators.fsns",
  "tensor2tensor.data_generators.function_docstring",
@@ -104,7 +107,6 @@
 ALL_MODULES = list(MODULES)
 
 
-
 def _is_import_err_msg(err_str, module):
  parts = module.split(".")
  suffixes = [".".join(parts[i:]) for i in range(len(parts))]