remove rogue keras import legacy

NREL · Apr 4, 2024 · 3d207e8 · 3d207e8
1 parent 6551856
commit 3d207e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pinn_spm_param/util/custom_activations.py b/pinn_spm_param/util/custom_activations.py
@@ -3,7 +3,7 @@
 import tensorflow as tf
 from conditionalDecorator import conditional_decorator
 from keras.backend import set_floatx
-from keras.layers import Activation
+from keras.layers import Activation, Layer
 from keras.utils import get_custom_objects
 
 set_floatx("float64")
@@ -26,7 +26,7 @@ def swish_activation(x):
     return x * tf.math.sigmoid(x)
 
 
-class Bswish(keras.layers.Layer):
+class Bswish(Layer):
     def __init__(self, num_outputs):
         super(Bswish, self).__init__()
         self.num_outputs = num_outputs