Skip to content

Commit d3276f1

Browse files
committed
refactor: move tacto frames into camera frames
1 parent b217c46 commit d3276f1

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

extensions/rcs_tacto/src/rcs_tacto/tacto_wrapper.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)