num_pre_message_passing_layers: <list[int]>
num_pre_message_passing_layers
0 <= num_pre_message_passing_layers <= 4
[0, 1, 2]