ValueError:

collection.find przyjmuje tylko jeden argument pozycyjny, a ty dajesz mu dwa.

Zmień swoje rozmowy tak, aby wyglądały tak:grades.find({"type": "homework", "student_id": i}) .