Update test/layers/basic.jl

CarloLucibello · web-flow · commit c4d562ce0ac8 · 2022-07-22T16:09:05.000+02:00
diff --git a/test/layers/basic.jl b/test/layers/basic.jl
@@ -68,7 +68,7 @@
     @testset "Flux restructure" begin
         chain = GNNChain(GraphConv(2=>2))
         params, restructure = Flux.destructure(chain)
-        restructure(params)
+        @test restructure(params) isa GNNChain
     end
 end
 
