Fix bug

Author: yuehhua

src/utils.jl

@@ -12,7 +12,7 @@ topk_index(y::Adjoint, k::Integer) = topk_index(y', k)
 ## Get feature with defaults
 
 get_feature(::Nothing, i) = nothing
-get_feature(::Fill{T,2,Axes}, i::Integer) where {T,Axes} = nothing
+get_feature(A::Fill{T,2,Axes}, i::Integer) where {T,Axes} = view(A, :, 1)
 get_feature(A::AbstractMatrix, i::Integer) = view(A, :, i)
 
 """

test/layers/msgpass.jl

@@ -2,6 +2,7 @@ in_channel = 10
 out_channel = 5
 num_V = 6
 num_E = 7
+T = Float32
 adj = [0. 1. 0. 0. 0. 0.;
        1. 0. 0. 1. 1. 1.;
        0. 0. 0. 0. 0. 1.;
@@ -16,8 +17,9 @@ NewLayer(m, n) = NewLayer(randn(m,n))
 
 (l::NewLayer)(fg) = propagate(l, fg, :add)
 
-X = Array(reshape(1:num_V*in_channel, in_channel, num_V))
+X = Array{T}(reshape(1:num_V*in_channel, in_channel, num_V))
 fg = FeaturedGraph(adj, X)
+fg.ef = Fill(zero(T), 0, 2num_E)
 
 l = NewLayer(out_channel, in_channel)
 

test/runtests.jl

@@ -2,6 +2,7 @@ using GeometricFlux
 using GeometricFlux.Datasets
 using Flux
 using Flux: @functor
+using FillArrays
 using GraphSignals
 using StaticArrays: @MMatrix, @MArray
 using LightGraphs: SimpleGraph, SimpleDiGraph, add_edge!, nv, ne