Explicitly replace "import tensorflow" with "tensorflow.compat.v1" for TF2.x migration

PiperOrigin-RevId: 293612141
This commit is contained in:
Deepmind Team
2020-02-06 17:45:36 +00:00
committed by Diego de Las Casas
parent 05dc76efa1
commit 48ba42f792
8 changed files with 8 additions and 8 deletions
+1 -1
View File
@@ -24,7 +24,7 @@ from absl import logging
import numpy as np
import six
import sonnet as snt
import tensorflow as tf # pylint: disable=g-explicit-tensorflow-version-import
import tensorflow.compat.v1 as tf # pylint: disable=g-explicit-tensorflow-version-import
from alphafold_casp13 import config_dict
from alphafold_casp13 import contacts_experiment
+1 -1
View File
@@ -18,7 +18,7 @@ import collections
import enum
import json
import tensorflow as tf # pylint: disable=g-explicit-tensorflow-version-import
import tensorflow.compat.v1 as tf # pylint: disable=g-explicit-tensorflow-version-import
_ProteinDescription = collections.namedtuple(
+1 -1
View File
@@ -15,7 +15,7 @@
"""Contact prediction convnet experiment example."""
from absl import logging
import tensorflow as tf # pylint: disable=g-explicit-tensorflow-version-import
import tensorflow.compat.v1 as tf # pylint: disable=g-explicit-tensorflow-version-import
from alphafold_casp13 import contacts_dataset
from alphafold_casp13 import contacts_network
+1 -1
View File
@@ -22,7 +22,7 @@ import os
import numpy as np
import six.moves.cPickle as pickle
import tensorflow as tf # pylint: disable=g-explicit-tensorflow-version-import
import tensorflow.compat.v1 as tf # pylint: disable=g-explicit-tensorflow-version-import
RR_FORMAT = """PFRMAT RR
+1 -1
View File
@@ -24,7 +24,7 @@ import os
from absl import app
from absl import flags
from absl import logging
import tensorflow as tf # pylint: disable=g-explicit-tensorflow-version-import
import tensorflow.compat.v1 as tf # pylint: disable=g-explicit-tensorflow-version-import
from alphafold_casp13 import distogram_io
from alphafold_casp13 import parsers
+1 -1
View File
@@ -16,7 +16,7 @@
import pickle
import tensorflow as tf # pylint: disable=g-explicit-tensorflow-version-import
import tensorflow.compat.v1 as tf # pylint: disable=g-explicit-tensorflow-version-import
def distance_histogram_dict(f):
+1 -1
View File
@@ -21,7 +21,7 @@ from absl import flags
from absl import logging
import numpy as np
import six
import tensorflow as tf # pylint: disable=g-explicit-tensorflow-version-import
import tensorflow.compat.v1 as tf # pylint: disable=g-explicit-tensorflow-version-import
from alphafold_casp13 import distogram_io
from alphafold_casp13 import parsers
+1 -1
View File
@@ -15,7 +15,7 @@
"""2D Resnet."""
from absl import logging
import tensorflow as tf # pylint: disable=g-explicit-tensorflow-version-import
import tensorflow.compat.v1 as tf # pylint: disable=g-explicit-tensorflow-version-import
from alphafold_casp13 import two_dim_convnet