Hey,

In render_annotation function you don't need a for loop (here) as there will always be one box in boxes (we are doing assertion checks for the same). And you are passing boxes[0] to view_points within the loop anyways. (here)

And, this time I wanna fix this by raising a PR, do let me know if I can process with that :)