Internal change

PiperOrigin-RevId: 410090466

official/nlp/tasks/__init__.py

@@ -12,3 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+"""TensorFlow Models NLP Tasks."""
+# pylint: disable=g-multiple-import
+from official.nlp.tasks.electra_task import ElectraPretrainConfig, ElectraPretrainTask
+from official.nlp.tasks.masked_lm import MaskedLMConfig, MaskedLMTask
+from official.nlp.tasks.question_answering import QuestionAnsweringConfig, QuestionAnsweringTask
+from official.nlp.tasks.sentence_prediction import SentencePredictionConfig, SentencePredictionTask
+from official.nlp.tasks.tagging import TaggingConfig, TaggingTask
+from official.nlp.tasks.translation import TranslationConfig, TranslationTask

tensorflow_models/nlp/__init__.py

@@ -13,4 +13,5 @@
 # limitations under the License.
 
 """TensorFlow Models NLP Libraries."""
+from official.nlp import tasks
 from official.nlp.modeling import *

tensorflow_models/tensorflow_models_test.py

@@ -28,6 +28,7 @@ class TensorflowModelsTest(tf.test.TestCase):
   def testNLPImport(self):
     _ = tfm.nlp.layers.TransformerEncoderBlock(
         num_attention_heads=2, inner_dim=10, inner_activation='relu')
+    _ = tfm.nlp.tasks.TaggingTask(params=tfm.nlp.tasks.TaggingConfig())
 
   def testCommonImports(self):
     _ = tfm.hyperparams.Config()