fix add tee

huggingface · Jul 3, 2024 · 291cca7 · 291cca7
1 parent 312c8b3
commit 291cca7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]