Fix default kwarg num_layers bug

bayesflow-org · May 24, 2024 · 08e5b31 · 08e5b31
1 parent ac7b66e
commit 08e5b31
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bayesflow/experimental/networks/coupling_flow/coupling_flow.py b/bayesflow/experimental/networks/coupling_flow/coupling_flow.py
@@ -39,7 +39,7 @@ def __init__(self, couplings: Sequence[AllInOneCoupling], base_distribution: Dis
     def all_in_one(
             cls,
             target_dim: int,
-            num_layers: 6,
+            num_layers=6,
             subnet_builder="default",
             transform="affine",
             permutation="fixed",