From f3fcb3c243084d07be4a6d48178889086215e86c Mon Sep 17 00:00:00 2001
From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= <francois@fleuret.org>
Date: Wed, 17 Jul 2024 06:49:18 +0200
Subject: [PATCH] Update.
---
quiz_machine.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
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(
--
2.39.5