Move engine registration from train command to build command