Update prim.jl

matbesancon · Feb 5, 2018 · 2d43a60 · 2d43a60
1 parent 4577920
commit 2d43a60
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/spanningtrees/prim.jl b/src/spanningtrees/prim.jl
@@ -1,5 +1,5 @@
 struct PrimHeapEntry{T<:Real}
-    edge::Edge
+    edge::AbstractEdge
     dist::T
 end
 
@@ -18,7 +18,7 @@ function prim_mst end
     distmx::AbstractMatrix = weights(g)
     )
     pq = Vector{PrimHeapEntry}()
-    mst = Vector{Edge}()
+    mst = Vector{AbstractEdge}()
     marked = zeros(Bool, nv(g))
 
     sizehint!(pq, ne(g))