Internal merge #160

PiperOrigin-RevId: 162389811

Author: Ryan Sepassi

tensor2tensor/bin/t2t-datagen

@@ -48,7 +48,7 @@ from tensor2tensor.data_generators import wiki
 from tensor2tensor.data_generators import wmt
 from tensor2tensor.data_generators import wsj_parsing
 from tensor2tensor.utils import registry
-from tensor2tensor.utils import registry_utils
+from tensor2tensor.utils import usr_dir
 
 import tensorflow as tf
 
@@ -280,7 +280,7 @@ def set_random_seed():
 
 def main(_):
   tf.logging.set_verbosity(tf.logging.INFO)
-  registry_utils.import_usr_dir(FLAGS.t2t_usr_dir)
+  usr_dir.import_usr_dir(FLAGS.t2t_usr_dir)
 
   # Calculate the list of problems to generate.
   problems = sorted(

tensor2tensor/bin/t2t-trainer

@@ -31,8 +31,8 @@ from __future__ import print_function
 
 # Dependency imports
 
-from tensor2tensor.utils import registry_utils
 from tensor2tensor.utils import trainer_utils as utils
+from tensor2tensor.utils import usr_dir
 
 import tensorflow as tf
 
@@ -49,7 +49,7 @@ flags.DEFINE_string("t2t_usr_dir", "",
 
 def main(_):
   tf.logging.set_verbosity(tf.logging.INFO)
-  registry_utils.import_usr_dir(FLAGS.t2t_usr_dir)
+  usr_dir.import_usr_dir(FLAGS.t2t_usr_dir)
   utils.log_registry()
   utils.validate_flags()
   utils.run(

tensor2tensor/utils/registry_utils.py renamed to tensor2tensor/utils/usr_dir.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Utilities for the t2t registry."""
+"""Utility to load code from an external user-supplied directory."""
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function