From dd201738cb96457e4b5ff11370efd2c6b86b168c Mon Sep 17 00:00:00 2001
From: ebuehrle <43623224+ebuehrle@users.noreply.github.com>
Date: Tue, 4 Oct 2022 20:51:19 +0200
Subject: [PATCH] when using IDM expert, update interaction graph in order to
 correctly display ego target agent

---
 src/baselines/rule_policies.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/baselines/rule_policies.py b/src/baselines/rule_policies.py
index d6c0913..4e361e3 100644
--- a/src/baselines/rule_policies.py
+++ b/src/baselines/rule_policies.py
@@ -172,6 +172,7 @@ class IDMRulePolicy(BaseAlgorithm):
 
             # Update environment interaction graph with leader
             self._env._env._graph._neighbor_dict={agent:[leader]}
+            self._env._update_graph = True
 
             delta_v = v_ego - v[leader, 0]
             d_des = self.d_min + self.tau * v_ego + v_ego * delta_v / (2* (self.a_max*self.b_pref)**0.5 )