add dummy coordinates to STNs from P matrices

rusandris · Aug 10, 2023 · 2ffbfd0 · 2ffbfd0
1 parent 437c76e
commit 2ffbfd0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/create_stn.jl b/src/create_stn.jl
@@ -164,14 +164,16 @@ function create_stn(P::AbstractMatrix;make_ergodic=false,verbose=false)
 	stn = MetaGraph(
         DiGraph(),
         Int64,
-        Dict{Symbol, Int64},
+        Dict{Symbol, Union{Int64,Float64}},
         Dict{Symbol, Float64},
         nothing,
         edge_data -> 1.0,
         0.0)
 
+	# add dummy coordinates
 	for v in 1:nr_vertices
-		stn[v] = Dict{Symbol, Int64}()
+		#stn[v] = Dict{Symbol, Int64}()
+		stn[v] = Dict(:x => v, :y => 0, :prob => 0.)
 	end
 
 	Q = calculate_weight_matrix(P)