Merge pull request #49 from jerryz123/collision_Fix

Fixed collision checking in collides_any_dynamic
BerkeleyAutomation · Apr 13, 2018 · 88e6773 · 88e6773
2 parents 949afb6 + c35fe01
commit 88e6773
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gym_urbandriving/state/state.py b/gym_urbandriving/state/state.py
@@ -218,7 +218,8 @@ def collides_any_dynamic(self, agentnum,type_of_agent = 'background_cars'):
         dynamic_collisions, static_collisions, _ = self.get_collisions()
 
         for coll in dynamic_collisions:
-            if (agentnum in coll) and (type_of_agent in coll):
+            id1, id2, t1, t2 = coll
+            if (agentnum, type_of_agent) in [(id1, t1), (id2, t2)]:
                 return True
 
         return False