From aeeefc1b58947ed34c01b51c638be874175c3844 Mon Sep 17 00:00:00 2001
From: "Albert R. Gnadt" <20693045+gnadt@users.noreply.github.com>
Date: Tue, 4 Mar 2025 00:42:17 -0500
Subject: [PATCH] Update check_zygote_type_stability()...

so that CI passes with Julia v1.10. Tested locally with Julia v1.10 & v1.11. Forces the expected type to be `Vector{Union{Nothing, @NamedTuple{X::Matrix{Float64}}}}` which is more robust for different package versions. There may be better solutions, but this seems to work.
---
 test/test_utils.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/test_utils.jl b/test/test_utils.jl
index 945c439ce..b2172a20a 100644
--- a/test/test_utils.jl
+++ b/test/test_utils.jl
@@ -104,7 +104,7 @@ function check_zygote_type_stability(f, args...; ctx=Zygote.Context())
     @inferred f(args...)
     @inferred Zygote._pullback(ctx, f, args...)
     out, pb = Zygote._pullback(ctx, f, args...)
-    @inferred pb(out)
+    @inferred collect(pb(out))
 end
 
 function test_ADs(