From 6eebae8b0728cbb8dc2f6f07c2577edd7e3390f6 Mon Sep 17 00:00:00 2001 From: Cedric Grothues Date: Tue, 24 Mar 2026 12:29:19 +0100 Subject: [PATCH] Fix geom type assignment --- .../pushing/state_based_global_pushing_env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/magbotsim/rl_envs/object_manipulation/pushing/state_based_global_pushing_env.py b/magbotsim/rl_envs/object_manipulation/pushing/state_based_global_pushing_env.py index e7beee1..039e65c 100644 --- a/magbotsim/rl_envs/object_manipulation/pushing/state_based_global_pushing_env.py +++ b/magbotsim/rl_envs/object_manipulation/pushing/state_based_global_pushing_env.py @@ -1025,7 +1025,7 @@ def _geoms(self, name: str) -> np.ndarray: break geom = self.model.geom(geom_id) - geoms[geom_idx, 0] = geom.type + geoms[geom_idx, 0] = geom.type.item() geoms[geom_idx, 1:3] = geom.size[:2] geoms[geom_idx, 3:5] = geom.pos[:2] geom_idx += 1