Revert "Modified keras import"

This reverts commit b3641a8.

alibi/datasets/tensorflow.py

@@ -1,8 +1,8 @@
 from typing import Tuple, Union
 
+import tensorflow.keras as keras
 import numpy as np
 
-from tensorflow import keras
 from alibi.utils.data import Bunch
 
 

alibi/explainers/backends/tensorflow/cfrl_base.py

@@ -9,7 +9,7 @@
 
 import numpy as np
 import tensorflow as tf
-from tensorflow import keras
+import tensorflow.keras as keras
 
 from alibi.explainers.backends.cfrl_base import CounterfactualRLDataset
 from alibi.models.tensorflow.actor_critic import Actor, Critic

alibi/explainers/tests/test_cfrl.py

@@ -5,7 +5,7 @@
 import numpy as np
 from numpy.testing import assert_allclose
 import tensorflow as tf
-from tensorflow import keras
+import tensorflow.keras as keras
 
 from alibi.explainers import CounterfactualRLTabular
 from alibi.explainers.backends.cfrl_base import get_hard_distribution

alibi/models/tensorflow/actor_critic.py

@@ -5,7 +5,8 @@
 """
 
 import tensorflow as tf
-from tensorflow import keras
+import tensorflow.keras as keras
+
 
 class Actor(keras.Model):
     """

alibi/models/tensorflow/autoencoder.py

@@ -5,7 +5,7 @@
 """
 
 import tensorflow as tf
-from tensorflow import keras
+import tensorflow.keras as keras
 from typing import List, Tuple, Union
 
 

alibi/models/tensorflow/cfrl_models.py

@@ -4,7 +4,7 @@
 """
 
 import tensorflow as tf
-from tensorflow import keras
+import tensorflow.keras as keras
 from typing import List
 
 

alibi/models/tensorflow/tests/test_autoencoder.py

@@ -1,6 +1,6 @@
 import numpy as np
 import pytest
-from tensorflow import keras
+import tensorflow.keras as keras
 from alibi.models.tensorflow.autoencoder import HeAE