From 291cca77e8cca3f148628e3420f308609904a1c4 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Wed, 3 Jul 2024 14:46:55 +0100 Subject: [PATCH] fix add tee --- gym_pusht/envs/pusht.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gym_pusht/envs/pusht.py b/gym_pusht/envs/pusht.py index 6fc13e5..73c9a35 100644 --- a/gym_pusht/envs/pusht.py +++ b/gym_pusht/envs/pusht.py @@ -511,8 +511,8 @@ def add_tee(space, position, angle, scale=30, color="LightSlateGray", mask=None) shape1.filter = pymunk.ShapeFilter(mask=mask) shape2.filter = pymunk.ShapeFilter(mask=mask) body.center_of_gravity = (shape1.center_of_gravity + shape2.center_of_gravity) / 2 - body.position = position body.angle = angle + body.position = position body.friction = 1 space.add(body, shape1, shape2) return body, [shape1, shape2]