Fix

Author: stojicnnnn

alignit/record.py

@@ -97,7 +97,10 @@ def main(cfg: RecordConfig):
     )
 
     for episode in range(cfg.episodes):
-        pose_start, pose_alignment_target = robot.reset(cfg)
+        if(cfg.robot_type == "real"):
+            pose_start, pose_alignment_target = robot.reset(cfg)
+        else:
+            pose_start, pose_alignment_target = robot.reset()
         trajectory = generate_spiral_trajectory(pose_start, cfg.trajectory)
         frames = []
         for pose in trajectory: