@@ -82,9 +82,9 @@ def reset(
8282 self .tacto_last_render = - 1 # Reset last render time
8383 colors , depths = self .tacto_sensor .render (self .model , self .data )
8484 for site , color , depth in zip (self .tacto_sites , colors , depths , strict = False ):
85- obs .setdefault ("tacto " , {}).setdefault (site , {}).setdefault ("rgb" , {})["data" ] = color
85+ obs .setdefault ("frames " , {}).setdefault (f"tactile_ { site } " , {}).setdefault ("rgb" , {})["data" ] = color
8686 if self .enable_depth :
87- obs .setdefault ("tacto " , {}).setdefault (site , {}).setdefault ("depth" , {})["data" ] = depth
87+ obs .setdefault ("frames " , {}).setdefault (f"tactile_ { site } " , {}).setdefault ("depth" , {})["data" ] = depth
8888 return obs , info
8989
9090 def step (self , action : dict [str , Any ]):
@@ -94,7 +94,9 @@ def step(self, action: dict[str, Any]):
9494 self .tacto_sensor .updateGUI (colors , depths ) if self .visualize else None
9595 self .tacto_last_render = self .data .time
9696 for site , color , depth in zip (self .tacto_sites , colors , depths , strict = False ):
97- obs .setdefault ("tacto " , {}).setdefault (site , {}).setdefault ("rgb" , {})["data" ] = color
97+ obs .setdefault ("frames " , {}).setdefault (f"tactile_ { site } " , {}).setdefault ("rgb" , {})["data" ] = color
9898 if self .enable_depth :
99- obs .setdefault ("tacto" , {}).setdefault (site , {}).setdefault ("depth" , {})["data" ] = depth
99+ obs .setdefault ("frames" , {}).setdefault (f"tactile_{ site } " , {}).setdefault ("depth" , {})[
100+ "data"
101+ ] = depth
100102 return obs , reward , done , truncated , info
0 commit comments