Internal change

PiperOrigin-RevId: 407664875

official/nlp/nhnet/trainer.py → official/projects/nhnet/trainer.py

@@ -17,18 +17,20 @@
 import os
 
 # Import libraries
+
 from absl import app
 from absl import flags
 from absl import logging
 from six.moves import zip
 import tensorflow as tf
+
 from official.common import distribute_utils
 from official.modeling.hyperparams import params_dict
-from official.nlp.nhnet import evaluation
-from official.nlp.nhnet import input_pipeline
-from official.nlp.nhnet import models
-from official.nlp.nhnet import optimizer
 from official.nlp.transformer import metrics as transformer_metrics
+from official.projects.nhnet import evaluation
+from official.projects.nhnet import input_pipeline
+from official.projects.nhnet import models
+from official.projects.nhnet import optimizer
 from official.utils.misc import keras_utils
 
 FLAGS = flags.FLAGS

official/nlp/nhnet/trainer_test.py → official/projects/nhnet/trainer_test.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Tests for official.nlp.nhnet.trainer."""
+"""Tests for official.projects.nhnet.trainer."""
 
 import os
 
@@ -24,8 +24,8 @@ import tensorflow as tf
 from tensorflow.python.distribute import combinations
 from tensorflow.python.distribute import strategy_combinations
 # pylint: enable=g-direct-tensorflow-import
-from official.nlp.nhnet import trainer
-from official.nlp.nhnet import utils
+from official.projects.nhnet import trainer
+from official.projects.nhnet import utils
 
 FLAGS = flags.FLAGS
 trainer.define_flags()

official/nlp/nhnet/utils.py → official/projects/nhnet/utils.py

@@ -20,7 +20,7 @@ import tensorflow as tf
 
 from official.modeling.hyperparams import params_dict
 from official.nlp.bert import configs
-from official.nlp.nhnet import configs as nhnet_configs
+from official.projects.nhnet import configs as nhnet_configs
 
 
 def get_bert_config_from_params(