Update src/NormalizingFlows.jl

zuhengxu · devmotion · web-flow · commit db4872b08c6a · 2025-03-05T08:40:33.000-08:00
Co-authored-by: David Widmann &lt;devmotion@users.noreply.github.com&gt;
diff --git a/src/NormalizingFlows.jl b/src/NormalizingFlows.jl
@@ -63,7 +63,8 @@ function train_flow(
         loss,
         θ_flat,
         re,
-        (rng, args...)...;
+        rng,
+        args...;
         max_iters=max_iters,
         optimiser=optimiser,
         kwargs...,
