Fix new pylint errors

nengo · May 10, 2022 · 0db0e77 · 0db0e77
1 parent 923b556
commit 0db0e77
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/keras_lmu/layers.py b/keras_lmu/layers.py
@@ -6,14 +6,13 @@
 import tensorflow as tf
 from packaging import version
 
+# pylint: disable=ungrouped-imports
 if version.parse(tf.__version__) < version.parse("2.6.0rc0"):
     from tensorflow.python.keras.layers.recurrent import DropoutRNNCellMixin
 else:
     from keras.layers.recurrent import DropoutRNNCellMixin
 
-if version.parse(tf.__version__) < version.parse(
-    "2.8.0rc0"
-):  # pylint: disable=ungrouped-imports
+if version.parse(tf.__version__) < version.parse("2.8.0rc0"):
     from tensorflow.keras.layers import Layer as BaseRandomLayer
 else:
     from keras.engine.base_layer import BaseRandomLayer