Assignment 6a - Section 1.9 - Output type inconsistency

[G0rocks]

2 different instructions in the code, assuming it should be a np.ndarray object since that's not a comment, see photo at bottom.
To prevent this, follow best practice of only writing each information point once and only once.
Not how I've removed the object type from the lines in the comments for all other inputs as we can see the type in the function parameters. This way, if you decide to update a type you only have to update it in one place and if you forget to update the comment it's no problem since it's only written once.