Fix _get_state function for StatefulLuxLayers

Author: simsurace

GNNLux/src/layers/conv.jl

@@ -1,9 +1,6 @@
 _getbias(ps) = hasproperty(ps, :bias) ? getproperty(ps, :bias) : false
 _getstate(st, name) = hasproperty(st, name) ? getproperty(st, name) : NamedTuple()
-_getstate(s::StatefulLuxLayer{true}) = s.st
-_getstate(s::StatefulLuxLayer{Static.True}) = s.st
-_getstate(s::StatefulLuxLayer{false}) = s.st_any
-_getstate(s::StatefulLuxLayer{Static.False}) = s.st_any
+_getstate(s::StatefulLuxLayer) = LuxCore.get_state(s)
 
 @doc raw"""
     GCNConv(in => out, σ=identity; [init_weight, init_bias, use_bias, add_self_loops, use_edge_weight])