Update prim.jl

sbromberger · Feb 6, 2018 · 677f6fa · 677f6fa
1 parent a0e9c47
commit 677f6fa
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/spanningtrees/prim.jl b/src/spanningtrees/prim.jl
@@ -13,10 +13,12 @@ distance matrix `distmx` using [Prim's algorithm](https://en.wikipedia.org/wiki/
 Return a vector of edges.
 """
 function prim_mst end
-@traitfn function prim_mst{T<:AbstractEdge, U<:Real, V, AG<:AbstractGraph{T}}(
+@traitfn function prim_mst{U<:Real, V, AG<:AbstractGraph{V}}(
     g::AG::(!IsDirected),
     distmx::AbstractMatrix{U} = weights(g)
-    )
+)
+
+    T = edgetype(g)
     pq = Vector{PrimHeapEntry{T, U}}()
     mst = Vector{T}()
     marked = zeros(Bool, nv(g))