Skip to content

Commit

Permalink
docs
Browse files Browse the repository at this point in the history
  • Loading branch information
danielward27 committed Feb 29, 2024
1 parent 68b0414 commit 4f61548
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion flowjax/flows.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ def coupling_flow(
key: Jax random number generator key.
base_dist: Base distribution, with ``base_dist.ndim==1``.
transformer: Bijection to be parameterised by conditioner. Defaults to
``_affine_transformer()``.
affine.
cond_dim: Dimension of conditioning variables. Defaults to None.
flow_layers: Number of coupling layers. Defaults to 8.
nn_width: Conditioner hidden layer size. Defaults to 50.
Expand Down

0 comments on commit 4f61548

Please sign in to comment.