[refactor] Move tn_bert/triviaqa from nlp/projects to official/projects

PiperOrigin-RevId: 419617646

official/nlp/projects/triviaqa/dataset.py → official/projects/triviaqa/dataset.py

@@ -23,7 +23,7 @@ import six
 import tensorflow as tf
 import tensorflow_datasets.public_api as tfds
 
-from official.nlp.projects.triviaqa import preprocess
+from official.projects.triviaqa import preprocess
 
 _CITATION = """
 @article{2017arXivtriviaqa,

official/nlp/projects/triviaqa/download_and_prepare.py → official/projects/triviaqa/download_and_prepare.py

@@ -21,7 +21,7 @@ from absl import logging
 import apache_beam as beam
 import tensorflow_datasets as tfds
 
-from official.nlp.projects.triviaqa import dataset  # pylint: disable=unused-import
+from official.projects.triviaqa import dataset  # pylint: disable=unused-import
 
 flags.DEFINE_integer('sequence_length', 4096, 'Max number of tokens.')
 

official/nlp/projects/triviaqa/evaluate.py → official/projects/triviaqa/evaluate.py

@@ -20,7 +20,7 @@ from absl import flags
 from absl import logging
 import tensorflow as tf
 
-from official.nlp.projects.triviaqa import evaluation
+from official.projects.triviaqa import evaluation
 
 flags.DEFINE_string('gold_path', None,
                     'Path to golden validation, i.e. wikipedia-dev.json.')

official/nlp/projects/triviaqa/inputs.py → official/projects/triviaqa/inputs.py

@@ -20,7 +20,7 @@ import tensorflow as tf
 import tensorflow_datasets as tfds
 
 from official.modeling import tf_utils
-from official.nlp.projects.triviaqa import dataset  # pylint: disable=unused-import
+from official.projects.triviaqa import dataset  # pylint: disable=unused-import
 
 
 def _flatten_dims(tensor: tf.Tensor,

official/nlp/projects/triviaqa/predict.py → official/projects/triviaqa/predict.py

@@ -27,9 +27,9 @@ import tensorflow_datasets as tfds
 
 import sentencepiece as spm
 from official.nlp.configs import encoders  # pylint: disable=unused-import
-from official.nlp.projects.triviaqa import evaluation
-from official.nlp.projects.triviaqa import inputs
-from official.nlp.projects.triviaqa import prediction
+from official.projects.triviaqa import evaluation
+from official.projects.triviaqa import inputs
+from official.projects.triviaqa import prediction
 
 flags.DEFINE_string('data_dir', None, 'TensorFlow Datasets directory.')
 

official/nlp/projects/triviaqa/preprocess.py → official/projects/triviaqa/preprocess.py

@@ -30,8 +30,8 @@ import numpy as np
 import tensorflow.io.gfile as gfile
 
 import sentencepiece as spm
-from official.nlp.projects.triviaqa import evaluation
-from official.nlp.projects.triviaqa import sentencepiece_pb2
+from official.projects.triviaqa import evaluation
+from official.projects.triviaqa import sentencepiece_pb2
 
 
 @dataclasses.dataclass

official/nlp/projects/triviaqa/train.py → official/projects/triviaqa/train.py

@@ -30,10 +30,10 @@ import tensorflow_datasets as tfds
 import sentencepiece as spm
 from official.nlp import optimization as nlp_optimization
 from official.nlp.configs import encoders
-from official.nlp.projects.triviaqa import evaluation
-from official.nlp.projects.triviaqa import inputs
-from official.nlp.projects.triviaqa import modeling
-from official.nlp.projects.triviaqa import prediction
+from official.projects.triviaqa import evaluation
+from official.projects.triviaqa import inputs
+from official.projects.triviaqa import modeling
+from official.projects.triviaqa import prediction
 
 flags.DEFINE_string('data_dir', None, 'Data directory for TensorFlow Datasets.')