Merge pull request #223 from SciML/fm/bugfix

weight init bugfix

Project.toml

@@ -1,7 +1,7 @@
 name = "ReservoirComputing"
 uuid = "7c2d2b1e-3dd4-11ea-355a-8f6a8116e294"
 authors = ["Francesco Martinuzzi"]
-version = "0.10.2"
+version = "0.10.3"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/ReservoirComputing.jl

@@ -67,6 +67,8 @@ function Predictive(prediction_data)
     Predictive(prediction_data, prediction_len)
 end
 
+__partial_apply(fn, inp) = fn$inp
+
 #fallbacks for initializers
 for initializer in (:rand_sparse, :delay_line, :delay_line_backward, :cycle_jumps,
     :simple_cycle, :pseudo_svd,
@@ -83,13 +85,13 @@ for initializer in (:rand_sparse, :delay_line, :delay_line_backward, :cycle_jump
         return $initializer(WeightInitializers._default_rng(), T, dims...; kwargs...)
     end
     @eval function ($initializer)(rng::AbstractRNG; kwargs...)
-        return WeightInitializers.__partial_apply($initializer, (rng, (; kwargs...)))
+        return __partial_apply($initializer, (rng, (; kwargs...)))
     end
     @eval function ($initializer)(rng::AbstractRNG,
             ::Type{T}; kwargs...) where {T <: $NType}
-        return WeightInitializers.__partial_apply($initializer, ((rng, T), (; kwargs...)))
+        return __partial_apply($initializer, ((rng, T), (; kwargs...)))
     end
-    @eval ($initializer)(; kwargs...) = WeightInitializers.__partial_apply(
+    @eval ($initializer)(; kwargs...) = __partial_apply(
         $initializer, (; kwargs...))
 end