Amperity interview question

Pair programming to for ML project.