Formatting fix

Author: avik-pal

ext/SparseDiffToolsSymbolicsExt.jl

@@ -3,7 +3,7 @@ module SparseDiffToolsSymbolicsExt
 using SparseDiffTools, Symbolics
 import SparseDiffTools: AbstractSparseADType
 
-function (alg::SymbolicsSparsityDetection)(ad::AbstractSparseADType, f, x; fx=nothing,
+function (alg::SymbolicsSparsityDetection)(ad::AbstractSparseADType, f, x; fx = nothing,
     kwargs...)
     fx = fx === nothing ? similar(f(x)) : dx
     f!(y, x) = (y .= f(x))

src/highlevel/finite_diff.jl

@@ -7,15 +7,15 @@ struct FiniteDiffJacobianCache{CO, CA, J, FX, X} <: AbstractMaybeSparseJacobianC
 end
 
 function sparse_jacobian_cache(fd::Union{AutoSparseFiniteDiff, AutoFiniteDiff},
-    sd::AbstractMaybeSparsityDetection, f, x; fx=nothing)
+    sd::AbstractMaybeSparsityDetection, f, x; fx = nothing)
     coloring_result = sd(fd, f, x)
     fx = fx === nothing ? similar(f(x)) : fx
     if coloring_result isa NoMatrixColoring
         cache = FiniteDiff.JacobianCache(x, fx)
         jac_prototype = nothing
     else
         cache = FiniteDiff.JacobianCache(x, fx; coloring_result.colorvec,
-            sparsity=coloring_result.jacobian_sparsity)
+            sparsity = coloring_result.jacobian_sparsity)
         jac_prototype = coloring_result.jacobian_sparsity
     end
     return FiniteDiffJacobianCache(coloring_result, cache, jac_prototype, fx, x)
@@ -29,7 +29,7 @@ function sparse_jacobian_cache(fd::Union{AutoSparseFiniteDiff, AutoFiniteDiff},
         jac_prototype = nothing
     else
         cache = FiniteDiff.JacobianCache(x, fx; coloring_result.colorvec,
-            sparsity=coloring_result.jacobian_sparsity)
+            sparsity = coloring_result.jacobian_sparsity)
         jac_prototype = coloring_result.jacobian_sparsity
     end
     return FiniteDiffJacobianCache(coloring_result, cache, jac_prototype, fx, x)

src/highlevel/forward_mode.jl

@@ -7,15 +7,16 @@ struct ForwardDiffJacobianCache{CO, CA, J, FX, X} <: AbstractMaybeSparseJacobian
 end
 
 function sparse_jacobian_cache(ad::Union{AutoSparseForwardDiff, AutoForwardDiff},
-    sd::AbstractMaybeSparsityDetection, f, x; fx=nothing)
+    sd::AbstractMaybeSparsityDetection, f, x; fx = nothing)
     coloring_result = sd(ad, f, x)
     fx = fx === nothing ? similar(f(x)) : fx
     if coloring_result isa NoMatrixColoring
         cache = ForwardDiff.JacobianConfig(f, x)
         jac_prototype = nothing
     else
-        cache = ForwardColorJacCache(f, x, __chunksize(ad); coloring_result.colorvec, dx=fx,
-            sparsity=coloring_result.jacobian_sparsity)
+        cache = ForwardColorJacCache(f, x, __chunksize(ad); coloring_result.colorvec,
+            dx = fx,
+            sparsity = coloring_result.jacobian_sparsity)
         jac_prototype = coloring_result.jacobian_sparsity
     end
     return ForwardDiffJacobianCache(coloring_result, cache, jac_prototype, fx, x)
@@ -29,7 +30,7 @@ function sparse_jacobian_cache(ad::Union{AutoSparseForwardDiff, AutoForwardDiff}
         jac_prototype = nothing
     else
         cache = ForwardColorJacCache(f!, x, __chunksize(ad); coloring_result.colorvec,
-            dx=fx, sparsity=coloring_result.jacobian_sparsity)
+            dx = fx, sparsity = coloring_result.jacobian_sparsity)
         jac_prototype = coloring_result.jacobian_sparsity
     end
     return ForwardDiffJacobianCache(coloring_result, cache, jac_prototype, fx, x)

src/highlevel/reverse_mode.jl

@@ -34,7 +34,7 @@ function sparse_jacobian!(J::AbstractMatrix, ad, cache::ReverseModeJacobianCache
 end
 
 function __sparse_jacobian_reverse_impl!(J::AbstractMatrix, ad, idx_vec,
-    cache::MatrixColoringResult, f, x, fx=nothing)
+    cache::MatrixColoringResult, f, x, fx = nothing)
     # If `fx` is `nothing` then assume `f` is not in-place
     @unpack colorvec, nz_rows, nz_cols = cache
     for c in 1:maximum(colorvec)

test/test_sparse_jacobian.jl

@@ -35,7 +35,6 @@ SPARSITY_DETECTION_ALGS = [JacPrototypeSparsityDetection(jac_prototype = J_spars
     @testset "sparse_jacobian: Out of Place" begin
         for difftype in (AutoSparseZygote(), AutoZygote(), AutoSparseForwardDiff(),
             AutoForwardDiff(), AutoSparseFiniteDiff(), AutoFiniteDiff())
-
             @testset "Cache & Reuse" begin
                 cache = sparse_jacobian_cache(difftype, sd, fdiff, x)
                 J = SparseDiffTools.__init_𝒥(cache)