train.py returns