diff --git a/mani_skill/utils/wrappers/record.py b/mani_skill/utils/wrappers/record.py index 683c3514d..3ffc84e2e 100644 --- a/mani_skill/utils/wrappers/record.py +++ b/mani_skill/utils/wrappers/record.py @@ -324,7 +324,9 @@ def capture_image(self, infos=None): img = common.to_numpy(img) if infos is not None: for i in range(len(img)): - info_item = {k: v[i] for k, v in infos.items()} + info_item = { + k: v if isinstance(v, float) else v[i] for k, v in infos.items() + } img[i] = put_info_on_image(img[i], info_item) if len(img.shape) > 3: if len(img) == 1: