Apply suggestions from code review

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

Author: yebai

ext/BijectorsMooncakeExt.jl

@@ -16,7 +16,10 @@ using Bijectors: find_alpha, ChainRulesCore
 @is_primitive(MinimalCtx, Tuple{typeof(find_alpha),P,P,Integer} where {P<:Base.IEEEFloat})
 
 function Mooncake.frule!!(
-    f::Mooncake.Dual{typeof(find_alpha)}, x::Mooncake.Dual{P}, y::Mooncake.Dual{P}, z::Mooncake.Dual{I}
+    f::Mooncake.Dual{typeof(find_alpha)},
+    x::Mooncake.Dual{P},
+    y::Mooncake.Dual{P},
+    z::Mooncake.Dual{I},
 ) where {P<:Base.IEEEFloat,I<:Integer}
     # Require that the integer is non-differentiable.
     if !(Mooncake.tangent(z) isa Mooncake.NoTangent)
@@ -25,8 +28,16 @@ function Mooncake.frule!!(
     end
     # Convert Mooncake.NoTangent to ChainRulesCore.NoTangent for the integer argument
     out, tangent_out = ChainRulesCore.frule(
-        (ChainRulesCore.NoTangent(), Mooncake.tangent(x), Mooncake.tangent(y), ChainRulesCore.NoTangent()),
-        find_alpha, Mooncake.primal(x), Mooncake.primal(y), Mooncake.primal(z)
+        (
+            ChainRulesCore.NoTangent(),
+            Mooncake.tangent(x),
+            Mooncake.tangent(y),
+            ChainRulesCore.NoTangent(),
+        ),
+        find_alpha,
+        Mooncake.primal(x),
+        Mooncake.primal(y),
+        Mooncake.primal(z),
     )
     return Mooncake.Dual(out, tangent_out)
 end