Internal change

PiperOrigin-RevId: 407664875

official/nlp/modeling/layers/multi_channel_attention_test.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Tests for nlp.nhnet.multi_channel_attention."""
+"""Tests for projects.nhnet.multi_channel_attention."""
 
 import numpy as np
 import tensorflow as tf

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

@@ -14,9 +14,8 @@
 
 """Common NHNet/Bert2Bert configuration."""
 
-from typing import List, Text
-
 import dataclasses
+from typing import List, Text
 
 from official.modeling.hyperparams import base_config
 

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

@@ -15,7 +15,7 @@
 """Tests for configs."""
 
 import tensorflow as tf
-from official.nlp.nhnet import configs
+from official.projects.nhnet import configs
 
 BERT2BERT_CONFIG = {
     "vocab_size": 30522,

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

@@ -12,14 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Tests for nlp.nhnet.decoder."""
+"""Tests for projects.nhnet.decoder."""
 
 import numpy as np
 import tensorflow as tf
 from official.nlp.modeling import layers
-from official.nlp.nhnet import configs
-from official.nlp.nhnet import decoder
-from official.nlp.nhnet import utils
+from official.projects.nhnet import configs
+from official.projects.nhnet import decoder
+from official.projects.nhnet import utils
 
 
 class DecoderTest(tf.test.TestCase):

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

@@ -15,15 +15,17 @@
 """Evaluation for Bert2Bert."""
 
 import os
+
 # Import libraries
+
 from absl import logging
 import numpy as np
 import tensorflow as tf
 
-from official.nlp.nhnet import input_pipeline
-from official.nlp.nhnet import models
 from official.nlp.transformer import metrics as metrics_v2
 from official.nlp.transformer.utils import metrics
+from official.projects.nhnet import input_pipeline
+from official.projects.nhnet import models
 
 
 def rouge_l_fscore(logits, labels):

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

@@ -24,9 +24,9 @@ from official.modeling.hyperparams import params_dict
 from official.nlp.modeling import networks
 from official.nlp.modeling.layers import multi_channel_attention
 from official.nlp.modeling.ops import beam_search
-from official.nlp.nhnet import configs
-from official.nlp.nhnet import decoder
-from official.nlp.nhnet import utils
+from official.projects.nhnet import configs
+from official.projects.nhnet import decoder
+from official.projects.nhnet import utils
 
 
 def embedding_linear(embedding_matrix, x):

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

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Tests for nlp.nhnet.models."""
+"""Tests for projects.nhnet.models."""
 
 import os
 
@@ -25,9 +25,9 @@ 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 configs
-from official.nlp.nhnet import models
-from official.nlp.nhnet import utils
+from official.projects.nhnet import configs
+from official.projects.nhnet import models
+from official.projects.nhnet import utils
 
 
 def all_strategy_combinations():

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

@@ -18,7 +18,7 @@ import os
 
 from absl import app
 from absl import flags
-from official.nlp.nhnet import raw_data_processor
+from official.projects.nhnet import raw_data_processor
 
 FLAGS = flags.FLAGS