From: François Fleuret <francois@fleuret.org>
Date: Wed, 17 Jul 2024 04:49:18 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=f3fcb3c243084d07be4a6d48178889086215e86c;p=culture.git

Update.
---

diff --git a/quiz_machine.py b/quiz_machine.py
index 14e8739..bbd1b7b 100755
--- 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(