On-the-job training for robots