Undo tf.compat import (aws#286)

* Undo import of tf compat v1

* remove tf.compat import

Author: rahul003

bin/sagemaker-containers/tensorflow/tf-train.py

@@ -1,7 +1,7 @@
 import argparse
 import numpy as np
 import sys
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 import time
 import uuid
 from tornasole.tensorflow import TornasoleHook, TornasoleOptimizer, SaveConfig

examples/tensorflow/scripts/mnist.py

@@ -1,7 +1,7 @@
 import argparse
 import numpy as np
 import random
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 import tornasole.tensorflow as ts
 
 parser = argparse.ArgumentParser()

examples/tensorflow/scripts/simple.py

@@ -1,6 +1,6 @@
 import argparse
 import numpy as np
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 import tornasole.tensorflow as ts
 import random
 

examples/tensorflow/scripts/train_imagenet_resnet_hvd.py

@@ -31,7 +31,7 @@
     from builtins import range
 except ImportError:
     pass
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 import numpy as np
 from tensorflow.contrib.image.python.ops import distort_image_ops
 from tensorflow.python.ops import data_flow_ops

tests/tensorflow/hooks/test_estimator_modes.py

@@ -10,7 +10,7 @@
 """
 
 import pytest
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 import numpy as np
 import shutil
 import os

tests/tensorflow/hooks/test_training_end.py

@@ -1,6 +1,6 @@
 from .utils import *
 from tornasole.tensorflow import reset_collections
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 from tornasole.core.access_layer.utils import has_training_ended
 import shutil
 import os

tests/tensorflow/hooks/test_weights_gradients.py

@@ -1,6 +1,6 @@
 from .utils import *
 from tornasole.tensorflow import reset_collections
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 from tornasole.core.config_constants import TORNASOLE_DEFAULT_COLLECTIONS_FILE_NAME
 from tornasole.core.json_config import TORNASOLE_CONFIG_FILE_PATH_ENV_STR
 import tornasole.tensorflow as ts

tests/tensorflow/hooks/test_write.py

@@ -1,4 +1,4 @@
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 import numpy as np
 from tornasole.tensorflow.hook import TornasoleHook
 from tensorflow.python.tools import inspect_checkpoint as chkp

tests/tensorflow/hooks/utils.py

@@ -1,4 +1,4 @@
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 import numpy as np
 from datetime import datetime
 import os

tests/tensorflow/test_tf_collections.py

@@ -1,5 +1,5 @@
 from tornasole.tensorflow import Collection, CollectionManager
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 
 
 def test_manager_export_load():

tornasole/tensorflow/collection.py

@@ -1,5 +1,5 @@
 from tensorflow.python.distribute import values
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 from tornasole.core.collection import Collection as BaseCollection, CollectionKeys
 from tornasole.core.collection_manager import CollectionManager as BaseCollectionManager
 

tornasole/tensorflow/hook.py

@@ -1,4 +1,4 @@
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 from .utils import *
 from .reductions import get_tensorflow_reduction
 from .collection import get_collection_manager

tornasole/tensorflow/optimizer.py

@@ -1,4 +1,4 @@
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 from .collection import *
 
 

tornasole/tensorflow/reductions.py

@@ -1,4 +1,4 @@
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 from tornasole.core.reduction_config import ALLOWED_REDUCTIONS, ALLOWED_NORMS
 
 

tornasole/tensorflow/utils.py

@@ -1,4 +1,4 @@
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 
 
 def node_name(n):