Fix jnp concatenate for sigma_bias

StevePny · Oct 31, 2023 · 04afde5 · 04afde5
1 parent 86e6bfd
commit 04afde5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dabench/model/_rc.py b/dabench/model/_rc.py
@@ -223,7 +223,7 @@ def update(self, r, u, A=None, Win=None):
 
         try:
             if self.sigma_bias != 0:
-                u = jnp.concatenate(([1.0], u))
+                u = jnp.concatenate((jnp.array([1.0]), u))
             p = A @ r.T + Win @ u
             q = self.leak_rate * jnp.tanh(p) + (1 - self.leak_rate) * r
         except TypeError as err: