Modified keras import

SeldonIO · Nov 29, 2024 · b3641a8 · b3641a8
1 parent cc47fc5
commit b3641a8
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 8 deletions.
diff --git a/alibi/datasets/tensorflow.py b/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
 
 

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

diff --git a/alibi/explainers/tests/test_cfrl.py b/alibi/explainers/tests/test_cfrl.py
@@ -5,7 +5,7 @@
 import numpy as np
 from numpy.testing import assert_allclose
 import tensorflow as tf
-import tensorflow.keras as keras
+from tensorflow import keras
 
 from alibi.explainers import CounterfactualRLTabular
 from alibi.explainers.backends.cfrl_base import get_hard_distribution

diff --git a/alibi/models/tensorflow/actor_critic.py b/alibi/models/tensorflow/actor_critic.py
@@ -5,8 +5,7 @@
 """
 
 import tensorflow as tf
-import tensorflow.keras as keras
-
+from tensorflow import keras
 
 class Actor(keras.Model):
     """

diff --git a/alibi/models/tensorflow/autoencoder.py b/alibi/models/tensorflow/autoencoder.py
@@ -5,7 +5,7 @@
 """
 
 import tensorflow as tf
-import tensorflow.keras as keras
+from tensorflow import keras
 from typing import List, Tuple, Union
 
 

diff --git a/alibi/models/tensorflow/cfrl_models.py b/alibi/models/tensorflow/cfrl_models.py
@@ -4,7 +4,7 @@
 """
 
 import tensorflow as tf
-import tensorflow.keras as keras
+from tensorflow import keras
 from typing import List
 
 

diff --git a/alibi/models/tensorflow/tests/test_autoencoder.py b/alibi/models/tensorflow/tests/test_autoencoder.py
@@ -1,6 +1,6 @@
 import numpy as np
 import pytest
-import tensorflow.keras as keras
+from tensorflow import keras
 from alibi.models.tensorflow.autoencoder import HeAE