This commit is contained in:
Jason Wen
2026-06-06 03:07:38 -04:00
parent 69749f400f
commit 41c7f23a74
+2 -1
View File
@@ -58,8 +58,9 @@ def make_split_input_queues(vision_input_shapes, policy_input_shapes, frame_skip
'traffic_convention': np.zeros(tc, dtype=np.float32),
'tfm': np.zeros((3, 3), dtype=np.float32),
'big_tfm': np.zeros((3, 3), dtype=np.float32),
'action_t': np.zeros(tc, dtype=np.float32),
}
if 'action_t' in policy_input_shapes:
npy['action_t'] = np.zeros(tc, dtype=np.float32)
handled = {'features_buffer', desire_key, 'traffic_convention', 'action_t'}
for key, shape in policy_input_shapes.items():