Add allocation id to training run model