Raffael-Kultyshev commited on
Commit
58c1274
·
1 Parent(s): 9e02de3

Add safety check for empty end_effector data

Browse files
Files changed (1) hide show
  1. app.py +13 -1
app.py CHANGED
@@ -60,7 +60,19 @@ def prepare_data():
60
  right_pos = {'x': [], 'y': [], 'z': []}
61
  right_rot = {'yaw': [], 'pitch': [], 'roll': []}
62
 
63
- frame_keys = sorted([int(k) for k in end_effector.keys() if k.isdigit()])
 
 
 
 
 
 
 
 
 
 
 
 
64
 
65
  for frame_idx in frame_keys:
66
  frame_key = str(frame_idx)
 
60
  right_pos = {'x': [], 'y': [], 'z': []}
61
  right_rot = {'yaw': [], 'pitch': [], 'roll': []}
62
 
63
+ if not end_effector:
64
+ return {
65
+ 'times': times,
66
+ 'left_pos': left_pos,
67
+ 'left_rot': left_rot,
68
+ 'right_pos': right_pos,
69
+ 'right_rot': right_rot
70
+ }
71
+
72
+ try:
73
+ frame_keys = sorted([int(k) for k in end_effector.keys() if str(k).isdigit()])
74
+ except:
75
+ frame_keys = []
76
 
77
  for frame_idx in frame_keys:
78
  frame_key = str(frame_idx)