trial_index,trial_type,IV_number_choice_only_trials,reward_inference_query,choice_inference_query,reward_prediction_query,choice_prediction_query
0,3,10,0,1,0,0
1,3,10,0,0,0,0
2,2,10,0,0,0,0
3,3,10,0,0,0,0
4,3,10,0,1,0,0
5,3,10,0,1,0,0
6,3,10,0,0,0,0
7,3,10,0,0,0,0
8,2,10,1,0,0,0
9,2,10,1,0,0,0
10,1,10,0,0,1,0
11,2,10,1,0,1,0
12,1,10,0,0,0,0
13,1,10,0,0,0,0
14,3,10,0,0,0,0
15,3,10,0,0,0,0
16,3,10,0,0,0,0
17,1,10,0,0,0,0
18,2,10,1,0,0,0
19,2,10,1,0,1,0
20,2,10,0,0,0,0
21,2,10,1,0,1,0
22,2,10,1,0,0,0
23,2,10,0,0,0,1
24,3,10,0,1,0,0
25,3,10,0,0,0,0
26,3,10,0,0,0,1
27,3,10,0,1,0,0
28,3,10,0,0,0,1
29,1,10,0,0,0,1
30,3,10,0,0,0,0
31,3,10,0,0,1,0
32,3,10,0,0,1,0
33,3,10,0,0,0,0
34,3,10,0,0,0,1
35,3,10,0,1,0,0
36,3,10,0,0,0,1
37,3,10,0,0,0,0
38,3,10,0,1,0,0
39,3,10,0,0,1,1