Replace training run ID attribute by entity relationship

Edited by Dominik Pilár