Explicitly replace "import tensorflow" with "tensorflow.compat.v1" fo…

…r TF2.x migration PiperOrigin-RevId: 295889122
xujinjin1 · Feb 19, 2020 · b32e50c · b32e50c
1 parent 4a0def2
commit b32e50c
Show file tree

Hide file tree

Showing 16 changed files with 18 additions and 18 deletions.
diff --git a/cluster_gcn/models.py b/cluster_gcn/models.py
@@ -17,7 +17,7 @@
 
 import layers
 import metrics
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 FLAGS = tf.flags.FLAGS
 

diff --git a/cluster_gcn/train.py b/cluster_gcn/train.py
@@ -19,7 +19,7 @@
 import models
 import numpy as np
 import partition_utils
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import utils
 
 tf.logging.set_verbosity(tf.logging.INFO)

diff --git a/dataset_analysis/bert_classifier.py b/dataset_analysis/bert_classifier.py
@@ -38,7 +38,7 @@
 from dataset_analysis.bert import tokenization
 import numpy as np
 import pandas as pd
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from tensorflow.contrib import data as contrib_data
 
 flags = tf.flags

diff --git a/dataset_analysis/learn_target_relations.py b/dataset_analysis/learn_target_relations.py
@@ -35,7 +35,7 @@
 from dataset_analysis.bert import tokenization
 import numpy as np
 import pandas as pd
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from tensorflow.contrib import data as contrib_data
 
 flags = tf.flags

diff --git a/igt_optimizer/cloud_tpu_resnet/resnet/imagenet_input.py b/igt_optimizer/cloud_tpu_resnet/resnet/imagenet_input.py
@@ -39,7 +39,7 @@
 import os
 # copybara:strip_begin
 from absl import flags
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from igt_optimizer.cloud_tpu_resnet.resnet import resnet_preprocessing
 from tensorflow.contrib import cloud as contrib_cloud
 from tensorflow.contrib import data as contrib_data

diff --git a/igt_optimizer/cloud_tpu_resnet/resnet/resnet_main.py b/igt_optimizer/cloud_tpu_resnet/resnet/resnet_main.py
@@ -41,7 +41,7 @@
 from absl import flags
 import absl.logging as _logging  # pylint: disable=unused-import
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from igt_optimizer import exp_igt_optimizer
 from igt_optimizer.cloud_tpu_resnet.hyperparameters import common_hparams_flags

diff --git a/l2tl/finetuning.py b/l2tl/finetuning.py
@@ -25,8 +25,8 @@
 from absl import flags
 import model
 import model_utils
-import tensorflow as tf
-from tensorflow.python.estimator import estimator
+import tensorflow.compat.v1 as tf
+from tensorflow.compat.v1.python.estimator import estimator
 import tensorflow_datasets as tfds
 
 flags.DEFINE_string(

diff --git a/l2tl/train_l2tl.py b/l2tl/train_l2tl.py
@@ -25,7 +25,7 @@
 from absl import flags
 import model
 import model_utils
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_datasets as tfds
 import tensorflow_probability as tfp
 

diff --git a/mol_dqn/chemgraph/dqn/deep_q_networks.py b/mol_dqn/chemgraph/dqn/deep_q_networks.py
@@ -30,7 +30,7 @@
 
 from six.moves import range
 from six.moves import zip
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from tensorflow.contrib import layers as contrib_layers
 from tensorflow.contrib import training as contrib_training
 

diff --git a/mol_dqn/chemgraph/dqn/run_dqn.py b/mol_dqn/chemgraph/dqn/run_dqn.py
@@ -40,8 +40,8 @@
 from rdkit.Chem import QED
 
 from six.moves import range
-import tensorflow as tf
-from tensorflow import gfile
+import tensorflow.compat.v1 as tf
+from tensorflow.compat.v1 import gfile
 
 from mol_dqn.chemgraph.dqn import deep_q_networks
 from mol_dqn.chemgraph.dqn import molecules as molecules_mdp

diff --git a/resolve_ref_exp_elements_ml/deeplab/mobilenet/mobilenet.py b/resolve_ref_exp_elements_ml/deeplab/mobilenet/mobilenet.py
@@ -24,7 +24,7 @@
 import copy
 import os
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from tensorflow.contrib import slim as contrib_slim
 
 slim = contrib_slim

diff --git a/resolve_ref_exp_elements_ml/deeplab/mobilenet/mobilenet_v2.py b/resolve_ref_exp_elements_ml/deeplab/mobilenet/mobilenet_v2.py
@@ -27,7 +27,7 @@
 
 import copy
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 import deeplab.mobilenet.conv_blocks as ops
 import deeplab.mobilenet.mobilenet as lib

diff --git a/schema_guided_dst/baseline/bert/modeling.py b/schema_guided_dst/baseline/bert/modeling.py
@@ -26,7 +26,7 @@
 import re
 import numpy as np
 import six
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 class BertConfig(object):

diff --git a/schema_guided_dst/baseline/bert/optimization.py b/schema_guided_dst/baseline/bert/optimization.py
@@ -21,7 +21,7 @@
 from __future__ import print_function
 
 import re
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 def create_optimizer(loss, init_lr, num_train_steps, num_warmup_steps, use_tpu):

diff --git a/schema_guided_dst/baseline/extract_schema_embedding.py b/schema_guided_dst/baseline/extract_schema_embedding.py
@@ -24,7 +24,7 @@
 import re
 
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from schema_guided_dst.baseline import data_utils
 from schema_guided_dst.baseline.bert import modeling

diff --git a/schema_guided_dst/baseline/train_and_predict.py b/schema_guided_dst/baseline/train_and_predict.py
@@ -28,7 +28,7 @@
 import os
 
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from schema_guided_dst import schema
 from schema_guided_dst.baseline import data_utils