From ccfd7a57298ebc1c373286479b4130ac27ce3b19 Mon Sep 17 00:00:00 2001 From: Michelle Casbon Date: Mon, 26 Feb 2024 14:36:22 -0800 Subject: [PATCH] Replace `tensorflow.compat.v2` import with `tensorflow` in vocabularies. PiperOrigin-RevId: 610530663 --- seqio/vocabularies.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/seqio/vocabularies.py b/seqio/vocabularies.py index bf9256ac..5061371a 100644 --- a/seqio/vocabularies.py +++ b/seqio/vocabularies.py @@ -22,7 +22,7 @@ from typing import Any, ClassVar, Dict, Iterable, Optional, Sequence, Union from absl import logging -import tensorflow.compat.v2 as tf +import tensorflow as tf import tensorflow_text as tf_text from sentencepiece import sentencepiece_model_pb2