change train edit run to write python configs