Update graph.jl

JuliaDynamics · Sep 6, 2024 · 6d373a4 · 6d373a4
1 parent 3be05e7
commit 6d373a4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/spaces/graph.jl b/src/spaces/graph.jl
@@ -68,8 +68,10 @@ random_position(model::ABM{<:GraphSpace}) = rand(abmrng(model), 1:nv(model))
 function remove_agent_from_space!(agent::AbstractAgent, model::ABM{<:GraphSpace})
     agentpos = agent.pos
     ids = ids_in_position(agentpos, model)
-    deleteat!(ids, findfirst(a -> a == agent.id, ids))
-    return model
+    agentid = findfirst(a -> a == agent.id, ids)
+    isnothing(agentid) && error(lazy"Tried to remove $(agent) from the space, but that agent is not on the space")
+    deleteat!(ids, agentid)
+    return agent
 end
 
 function add_agent_to_space!(agent::AbstractAgent, model::ABM{<:GraphSpace})