I wouldn’t blame the specific engineer for this too much (for asking you to write down the code). If Amazon is the same as Google, then there is a lot done to standardise interview scoring. Part of the guidance given to interviewers is to make sure that the candidate has produced actual “working” code. The code is included in the interview feedback that the interviewer writes.

Source: Interview trained at Google