diff --git a/cluster_gcn/models.py b/cluster_gcn/models.py old mode 100755 new mode 100644 index 75b3db608fd4..e67f05c51c5e --- 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 old mode 100755 new mode 100644 index 5f5bf806bab7..f7a08e38cf1b --- 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 index 62c268b6d6b0..4bc395e25092 100644 --- 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 index 483dd326a6cd..aaea03f36213 100644 --- 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 index 65c01e60449f..16658e975a6c 100644 --- 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 index 529ac99cf712..2998dfdac390 100644 --- 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 index 907b22af3b68..0b447bc5248a 100644 --- 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 index 2fbe91d8ccb9..1078c52fb76a 100644 --- 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 index 6f56a15a85de..2e9f5a2e0d09 100644 --- 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 index 6a325be8dfa7..a8349033af3c 100644 --- 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 index 4e858ed35ca0..533e42973d90 100644 --- 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 index 50894b59fc5f..cd1cdcd89936 100644 --- 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 index f622c66027e2..9a3c00e28601 100644 --- 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 index d3d25750b255..136d8476c3a2 100644 --- 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 index 84e5912bdd8c..d5aa8beb5952 100644 --- 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 index 22083cd0753c..454d51120daa 100644 --- 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