format

SciML · Feb 27, 2024 · 2165795 · 2165795
1 parent fbb0ceb
commit 2165795
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/ReservoirComputing.jl b/src/ReservoirComputing.jl
@@ -94,7 +94,8 @@ for initializer in (:rand_sparse, :delay_line, :delay_line_backward, :cycle_jump
             ::Type{T}; kwargs...) where {T <: $NType}
         return WeightInitializers.__partial_apply($initializer, ((rng, T), (; kwargs...)))
     end
-    @eval ($initializer)(; kwargs...) = WeightInitializers.__partial_apply($initializer, (; kwargs...))
+    @eval ($initializer)(; kwargs...) = WeightInitializers.__partial_apply(
+        $initializer, (; kwargs...))
 end
 
 #general

diff --git a/src/esn/esn_reservoirs.jl b/src/esn/esn_reservoirs.jl
@@ -25,9 +25,8 @@ function rand_sparse(rng::AbstractRNG,
         radius = T(1.0),
         sparsity = T(0.1),
         std = T(1.0)) where {T <: Number}
-
-    lcl_sparsity = T(1)-sparsity #consistency with current implementations
-    reservoir_matrix = sparse_init(rng, T, dims...; sparsity=lcl_sparsity, std=std)
+    lcl_sparsity = T(1) - sparsity #consistency with current implementations
+    reservoir_matrix = sparse_init(rng, T, dims...; sparsity = lcl_sparsity, std = std)
     rho_w = maximum(abs.(eigvals(reservoir_matrix)))
     reservoir_matrix .*= radius / rho_w
     if Inf in unique(reservoir_matrix) || -Inf in unique(reservoir_matrix)