from pfp.common.se3_utils import init_random_traj_th
from pfp.common.visualization import RerunViewer as RV

B = 2
T = 10
traj = init_random_traj_th(B, T)
# Vis first traj
RV("random_traj")
RV.add_traj("random_traj", traj[0].cpu().numpy())

print("Done.")