ValueError:

collection.find prende solo un argomento posizionale e tu gliene dai due.

Modifica le tue chiamate in modo che assomiglino alle seguenti:grades.find({"type": "homework", "student_id": i}) .