From 0a9fe2e206d1246e902b0fee4ed62d10d3502863 Mon Sep 17 00:00:00 2001
From: MartinuzziFrancesco <martinuzzi.francesco@gmail.com>
Date: Sun, 28 Aug 2022 20:21:48 +0200
Subject: [PATCH] some fixes

---
 src/ReservoirComputing.jl        | 2 +-
 src/esn/esn_reservoir_drivers.jl | 9 ++++++---
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/ReservoirComputing.jl b/src/ReservoirComputing.jl
index d4dec0da..37863e45 100644
--- a/src/ReservoirComputing.jl
+++ b/src/ReservoirComputing.jl
@@ -19,7 +19,7 @@ export AbstractLayer, create_layer
 export WeightedLayer, DenseLayer, SparseLayer, MinimumLayer, InformedLayer, NullLayer
 export BernoulliSample, IrrationalSample
 export GaussianProcess
-export AbstractReservoir, create_reservoir
+export AbstractReservoir, create_reservoir, create_states
 export RandSparseReservoir, PseudoSVDReservoir, DelayLineReservoir
 export DelayLineBackwardReservoir, SimpleCycleReservoir, CycleJumpsReservoir, NullReservoir
 export RNN, MRNN, GRU, GRUParams, FullyGated, Variant1, Variant2, Variant3, Minimal
diff --git a/src/esn/esn_reservoir_drivers.jl b/src/esn/esn_reservoir_drivers.jl
index e339a3c3..6467df86 100644
--- a/src/esn/esn_reservoir_drivers.jl
+++ b/src/esn/esn_reservoir_drivers.jl
@@ -1,9 +1,12 @@
 abstract type AbstractReservoirDriver end
 
 """
-    create_states(
-        reservoir_driver::AbstractReservoirDriver, train_data, reservoir_matrix,input_matrix
-    )
+    create_states(reservoir_driver::AbstractReservoirDriver,
+                       train_data,
+                       washout,
+                       reservoir_matrix,
+                       input_matrix,
+                       bias_vector)
 
 Return the trained ESN states according to the given driver.
 """