fix: some additional small fixes

Author: MartinuzziFrancesco

src/extensions/reca.jl

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

src/inits/esn_inits.jl

@@ -306,17 +306,17 @@ function informed_init(rng::AbstractRNG, ::Type{T}, dims::Integer...;
                             for jdx in axes(input_matrix, 1)])
         random_row_idx = idxs[DeviceAgnostic.rand(rng, T, 1:end)]
         random_clm_idx = range(1, state_size; step = 1)[DeviceAgnostic.rand(
-            rng, T, 1:end)]
+            rng, T, 1:length(idxs))]
         input_matrix[random_row_idx, random_clm_idx] = (DeviceAgnostic.rand(rng, T) -
                                                         T(0.5)) .* (T(2) * T(scaling))
     end
 
     for idx in 1:num_for_model
         idxs = findall(Bool[zero_connections .== input_matrix[jdx, :]
                             for jdx in axes(input_matrix, 1)])
-        random_row_idx = idxs[DeviceAgnostic.rand(rng, T, 1:end)]
+        random_row_idx = idxs[DeviceAgnostic.rand(rng, T, 1:length(idxs))]
         random_clm_idx = range(state_size + 1, in_size; step = 1)[DeviceAgnostic.rand(
-            rng, T, 1:end)]
+            rng, T, 1:length(idxs))]
         input_matrix[random_row_idx, random_clm_idx] = (DeviceAgnostic.rand(rng, T) -
                                                         T(0.5)) .* (T(2) * T(scaling))
     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::Any
+    activation
     in_dims <: IntegerType
     out_dims <: IntegerType
-    init_weight::Any
-    init_bias::Any
+    init_weight
+    init_bias
     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::Any
+    activation
     in_dims <: IntegerType
     out_dims <: IntegerType
-    init_bias::Any
-    init_reservoir::Any
-    init_input::Any
+    init_bias
+    init_reservoir
+    init_input
     #init_feedback::F
-    init_state::Any
-    leak_coefficient::Any
+    init_state
+    leak_coefficient
     use_bias <: StaticBool
 end
 

src/layers/lux_layers.jl

@@ -16,7 +16,7 @@ next step.
 
 ## Arguments
 
-- `cell`: Any `AbstractReservoirRecurrentCell` (e.g. [`ESNCell`](@ref)).
+- `cell`: `AbstractReservoirRecurrentCell` (e.g. [`ESNCell`](@ref)).
 
 ## States
 
@@ -118,7 +118,7 @@ updated states for every layer.
 """
 @concrete struct ReservoirChain <: AbstractLuxWrapperLayer{:layers}
     layers <: NamedTuple
-    name::Any
+    name
 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::Any
-    states_modifiers::Any
-    readout::Any
+    cells
+    states_modifiers
+    readout
 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::Any
-    knowledge_model::Any
-    states_modifiers::Any
-    readout::Any
+    cell
+    knowledge_model
+    states_modifiers
+    readout
 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::Any
-    states_modifiers::Any
-    readout::Any
+    reservoir
+    states_modifiers
+    readout
 end
 
 function initialparameters(rng::AbstractRNG, rc::ReservoirComputer)