From 9e42c8cda2a3d5ca4d496f17e1218ada98ce16f0 Mon Sep 17 00:00:00 2001
From: Simone Carlo Surace <51025924+simsurace@users.noreply.github.com>
Date: Tue, 28 Oct 2025 00:41:53 +0100
Subject: [PATCH 1/2] Restrict LuxTestUtils to v1

The GNNLux tests currently fail because of breaking changes in LuxTestUtils v2 (or so it seems from a casual look). Until someone has time to upgrade the GNNLux tests, I propose setting a compat bound in order to see if anything is truly broken.
---
 GNNLux/test/Project.toml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/GNNLux/test/Project.toml b/GNNLux/test/Project.toml
index d651808bf..75b82f42d 100644
--- a/GNNLux/test/Project.toml
+++ b/GNNLux/test/Project.toml
@@ -20,3 +20,6 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 TestItemRunner = "f8b46487-2199-4994-9208-9a1283c18c0a"
 TestItems = "1c621080-faea-4a02-84b6-bbd5e436b8fe"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
+
+[compat]
+LuxTestUtils = "1"

From 012b23c179f50746f22364c5aa0172058d2b971d Mon Sep 17 00:00:00 2001
From: Simone Carlo Surace <simsurace@gmx.net>
Date: Tue, 28 Oct 2025 22:31:30 +0100
Subject: [PATCH 2/2] Fix `_get_state` function for `StatefulLuxLayer`s

---
 GNNLux/src/layers/conv.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
index 63c4f90b4..3b70747c8 100644
--- a/GNNLux/src/layers/conv.jl
+++ b/GNNLux/src/layers/conv.jl
@@ -1,8 +1,8 @@
 _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{Val{true}}) = s.st
 _getstate(s::StatefulLuxLayer{Static.True}) = s.st
-_getstate(s::StatefulLuxLayer{false}) = s.st_any
+_getstate(s::StatefulLuxLayer{Val{false}}) = s.st_any
 _getstate(s::StatefulLuxLayer{Static.False}) = s.st_any
 
 @doc raw"""