Internal change

PiperOrigin-RevId: 432292375

Internal change
PiperOrigin-RevId: 432292375
b136d845 · A. Unique TensorFlower · c3079af6 · b136d845
Commit b136d845 authored Mar 03, 2022 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

official/nlp/modeling/layers/text_layers.py official/nlp/modeling/layers/text_layers.py +2 -0

No files found.
--- a/official/nlp/modeling/layers/text_layers.py
+++ b/official/nlp/modeling/layers/text_layers.py
@@ -20,8 +20,10 @@ from absl import logging
 import tensorflow as tf

 try:
+  # pytype: disable=import-error
  import tensorflow_text as text
  from tensorflow_text.python.ops import bert_tokenizer
+  # pytype: enable=import-error
 except ImportError:
  text = None
  bert_tokenizer = None