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]