Skip to content

Commit 67a1575

Browse files
authored
Fix GNNLux test failures (#626)
* 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. * Fix `_get_state` function for `StatefulLuxLayer`s
1 parent a4b27ae commit 67a1575

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

GNNLux/src/layers/conv.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
_getbias(ps) = hasproperty(ps, :bias) ? getproperty(ps, :bias) : false
22
_getstate(st, name) = hasproperty(st, name) ? getproperty(st, name) : NamedTuple()
3-
_getstate(s::StatefulLuxLayer{true}) = s.st
3+
_getstate(s::StatefulLuxLayer{Val{true}}) = s.st
44
_getstate(s::StatefulLuxLayer{Static.True}) = s.st
5-
_getstate(s::StatefulLuxLayer{false}) = s.st_any
5+
_getstate(s::StatefulLuxLayer{Val{false}}) = s.st_any
66
_getstate(s::StatefulLuxLayer{Static.False}) = s.st_any
77

88
@doc raw"""

GNNLux/test/Project.toml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,3 +20,6 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
2020
TestItemRunner = "f8b46487-2199-4994-9208-9a1283c18c0a"
2121
TestItems = "1c621080-faea-4a02-84b6-bbd5e436b8fe"
2222
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
23+
24+
[compat]
25+
LuxTestUtils = "1"

0 commit comments

Comments
 (0)