import train python script