Fix

Co-authored-by: Carlo Lucibello <[email protected]>
JuliaGraphs · Aug 19, 2024 · 7d78200 · 7d78200
1 parent e4b6f80
commit 7d78200
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/GNNLux/src/layers/temporalconv.jl b/GNNLux/src/layers/temporalconv.jl
@@ -12,8 +12,8 @@ function (r::StatefulRecurrentCell)(g, x::AbstractMatrix, ps, st::NamedTuple)
 end
 
 function (r::StatefulRecurrentCell)(g, x::AbstractVector, ps, st::NamedTuple)
-    (out, carry), st = applyrecurrentcell(r.cell, g, first(x), ps, st.cell, st.carry)
-    for xᵢ in x[(begin + 1):end]
+    st, carry = st.cell, st.carry
+    for xᵢ in x
         (out, carry), st = applyrecurrentcell(r.cell, g, xᵢ, ps, st, carry)
     end
     return out, (; cell=st, carry)