Update.

diff --git a/quiz_machine.py b/quiz_machine.py
index 14e8739..bbd1b7b 100755 (executable)
--- a/quiz_machine.py
+++ b/quiz_machine.py
@@ -341,7 +341,7 @@ class QuizMachine:
                 ) < w_quizzes.size(0)
                 i = torch.randperm(quizzes.size(0), device=quizzes.device)
 
-                return quizzes[i], type_w[i]
+                return quizzes[i], from_w[i]
 
             else:
                 return w_quizzes, torch.full(