Remove compat.v2 import.

PiperOrigin-RevId: 354439331

official/recommendation/create_ncf_data.py

@@ -14,17 +14,13 @@
 # ==============================================================================
 """Binary to generate training/evaluation dataset for NCF model."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import json
 
 # pylint: disable=g-bad-import-order
 # Import libraries
 from absl import app
 from absl import flags
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 # pylint: enable=g-bad-import-order
 
 from official.recommendation import movielens

official/recommendation/ncf_input_pipeline.py

@@ -14,15 +14,10 @@
 # ==============================================================================
 """NCF model input pipeline."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import functools
 
 # pylint: disable=g-bad-import-order
-
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 # pylint: enable=g-bad-import-order
 
 from official.recommendation import constants as rconst

official/recommendation/ncf_keras_main.py

@@ -18,10 +18,6 @@ The NeuMF model assembles both MF and MLP models under the NCF framework. Check
 `neumf_model.py` for more details about the models.
 """
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import json
 import os
 
@@ -30,7 +26,7 @@ import os
 from absl import app
 from absl import flags
 from absl import logging
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 # pylint: enable=g-bad-import-order
 
 from official.common import distribute_utils

official/staging/training/grad_utils.py

@@ -14,14 +14,9 @@
 # ==============================================================================
 """Some gradient util functions to help users writing custom training loop."""
 
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
-
 from absl import logging
 
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 
 
 def _filter_grads(grads_and_vars):

official/vision/beta/serving/export_saved_model_lib.py

@@ -17,7 +17,7 @@ r"""Vision models export utility function for serving/inference."""
 
 import os
 
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 
 from official.core import train_utils
 from official.vision.beta import configs