Skip to content

Commit c44333d

Browse files
committed
Revert layer type for coupling flow [skip ci]
1 parent d42fe51 commit c44333d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bayesflow/networks/coupling_flow/coupling_flow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ class CouplingFlow(InferenceNetwork):
4040

4141
def __init__(
4242
self,
43-
subnet: str | type | keras.Layer = "mlp",
43+
subnet: str | type = "mlp",
4444
depth: int = 6,
4545
transform: str = "affine",
4646
permutation: str | None = "random",
@@ -68,7 +68,7 @@ def __init__(
6868
6969
Parameters
7070
----------
71-
subnet : str, type, or keras.Layer, optional
71+
subnet : str or type optional
7272
Architecture for the transformation network. Can be "mlp", a custom network class, or
7373
a Layer object, e.g., `bayesflow.networks.MLP(widths=[32, 32])`. Default is "mlp".
7474
depth : int, optional

0 commit comments

Comments
 (0)