Simplify import structure

bayesflow-org · Jun 12, 2024 · 7280f40 · 7280f40
1 parent 84ed906
commit 7280f40
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 13 deletions.
diff --git a/bayesflow/experimental/networks/coupling_flow/permutations/fixed_permutation.py b/bayesflow/experimental/networks/coupling_flow/permutations/fixed_permutation.py
@@ -1,8 +1,6 @@
 
 import keras
-from keras.saving import (
-    register_keras_serializable,
-)
+from keras.saving import register_keras_serializable
 
 from bayesflow.experimental.types import Shape, Tensor
 from ..invertible_layer import InvertibleLayer

diff --git a/bayesflow/experimental/networks/coupling_flow/permutations/orthogonal.py b/bayesflow/experimental/networks/coupling_flow/permutations/orthogonal.py
@@ -1,8 +1,6 @@
 
 from keras import ops
-from keras.saving import (
-    register_keras_serializable,
-)
+from keras.saving import register_keras_serializable
 
 from bayesflow.experimental.types import Shape, Tensor
 from ..invertible_layer import InvertibleLayer

diff --git a/bayesflow/experimental/networks/coupling_flow/permutations/random.py b/bayesflow/experimental/networks/coupling_flow/permutations/random.py
@@ -1,8 +1,6 @@
 
 import keras
-from keras.saving import (
-    register_keras_serializable,
-)
+from keras.saving import register_keras_serializable
 
 from bayesflow.experimental.types import Shape
 from .fixed_permutation import FixedPermutation

diff --git a/bayesflow/experimental/networks/coupling_flow/permutations/swap.py b/bayesflow/experimental/networks/coupling_flow/permutations/swap.py
@@ -1,8 +1,6 @@
 
 import keras
-from keras.saving import (
-    register_keras_serializable,
-)
+from keras.saving import register_keras_serializable
 
 from bayesflow.experimental.types import Shape
 from .fixed_permutation import FixedPermutation
@@ -26,5 +24,5 @@ def build(self, input_shape: Shape) -> None:
             shape=(input_shape[-1],),
             initializer=keras.initializers.Constant(inverse_indices),
             trainable=False,
-            dtype = "int32"
+            dtype="int32"
         )