fix: some additional small fixes

Author: MartinuzziFrancesco

src/extensions/reca.jl

@@ -128,7 +128,7 @@ and [Yilmaz2014](@cite).
 
 """
 @concrete struct RECACell <: AbstractReservoirRecurrentCell
-    automaton
+    automaton::Any
     enc <: RandomMaps
 end
 

src/layers/basic.jl

@@ -51,11 +51,11 @@ before this layer (logically inserting a [`Collect`](@ref) right before it).
   which is usually unintended.
 """
 @concrete struct LinearReadout <: AbstractReservoirTrainableLayer
-    activation
+    activation::Any
     in_dims <: IntegerType
     out_dims <: IntegerType
-    init_weight
-    init_bias
+    init_weight::Any
+    init_bias::Any
     use_bias <: StaticBool
     include_collect <: StaticBool
 end

src/layers/esn_cell.jl

@@ -66,15 +66,15 @@ Created by `initialstates(rng, esn)`:
   - `rng`: a replicated RNG used to sample initial hidden states when needed.
 """
 @concrete struct ESNCell <: AbstractEchoStateNetworkCell
-    activation
+    activation::Any
     in_dims <: IntegerType
     out_dims <: IntegerType
-    init_bias
-    init_reservoir
-    init_input
+    init_bias::Any
+    init_reservoir::Any
+    init_input::Any
     #init_feedback::F
-    init_state
-    leak_coefficient
+    init_state::Any
+    leak_coefficient::Any
     use_bias <: StaticBool
 end
 

src/layers/lux_layers.jl

@@ -118,7 +118,7 @@ updated states for every layer.
 """
 @concrete struct ReservoirChain <: AbstractLuxWrapperLayer{:layers}
     layers <: NamedTuple
-    name
+    name::Any
 end
 
 function ReservoirChain(xs...; name = nothing)

src/models/esn_deep.jl

@@ -115,9 +115,9 @@ Composition:
 
 """
 @concrete struct DeepESN <: AbstractEchoStateNetwork{(:cells, :states_modifiers, :readout)}
-    cells
-    states_modifiers
-    readout
+    cells::Any
+    states_modifiers::Any
+    readout::Any
 end
 
 function DeepESN(in_dims::IntegerType,

src/models/esn_hybrid.jl

@@ -79,10 +79,10 @@ Created by `initialstates(rng, hesn)`:
 """
 @concrete struct HybridESN <: AbstractEchoStateNetwork{(
     :cell, :states_modifiers, :readout, :knowledge_model)}
-    cell
-    knowledge_model
-    states_modifiers
-    readout
+    cell::Any
+    knowledge_model::Any
+    states_modifiers::Any
+    readout::Any
 end
 
 function HybridESN(km,

src/reservoircomputer.jl

@@ -31,9 +31,9 @@ features, and install trained readout weights.
 """
 @concrete struct ReservoirComputer <:
                  AbstractReservoirComputer{(:reservoir, :states_modifiers, :readout)}
-    reservoir
-    states_modifiers
-    readout
+    reservoir::Any
+    states_modifiers::Any
+    readout::Any
 end
 
 function initialparameters(rng::AbstractRNG, rc::ReservoirComputer)