I'd say "мы ждали поезд целый день", "ждали поезда" sounds weird.