Train flux struct with list of models

Yes, you’re right, my bd. However, removing it didnt’ do anything.