diff --git a/src/dora_openarm_classifier/main.py b/src/dora_openarm_classifier/main.py index f052c14..3507890 100644 --- a/src/dora_openarm_classifier/main.py +++ b/src/dora_openarm_classifier/main.py @@ -151,7 +151,10 @@ def main(): continue if event["id"] == "position": - last_pos = event["value"].to_numpy(zero_copy_only=False) + value = event["value"] + if isinstance(value, pa.StructArray): + value = value.field("qpos") + last_pos = value.to_numpy(zero_copy_only=False) last_right_pos = last_pos[:8] last_left_pos = last_pos[9:] continue