Scheduled Commit

scripts/eval/trex_ablation_eval.py

@@ -59,7 +59,10 @@ VIDEO_KEYS = ("video.head_left", "video.left_wrist", "video.right_wrist")
 
 def _column(episode: TrexEpisode, name: str, dtype=np.float32) -> np.ndarray:
     values = episode.table.column(name).to_numpy(zero_copy_only=False)
-    return np.stack([np.asarray(row, dtype=dtype) for row in values])
+    values = np.asarray(values)
+    while values.dtype == object:
+        values = np.stack([np.stack(row) for row in values])
+    return values.astype(dtype)
 
 
 def _sample(timestamps, anchor, offsets, rate):