Iterator

ocadotechnology · Feb 27, 2024 · 958d1fa · 958d1fa
1 parent 30cb093
commit 958d1fa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/game/migrations/0090_add_missing_model_solutions.py b/game/migrations/0090_add_missing_model_solutions.py
@@ -24,7 +24,7 @@ def add_missing_model_solutions(apps: Apps, *args):
 
     for level_name, model_solution in model_solutions.items():
         count = Level.objects.filter(name=level_name).update(
-            model_solution=model_solution
+            model_solution=model_solution, disable_algorithm_score=False
         )
         assert count == 1
 
@@ -46,9 +46,9 @@ def add_missing_model_solutions(apps: Apps, *args):
             "91",
         ],
         score=10,
-    )
+    ).prefetch_related("level")
 
-    for attempt in attempts:
+    for attempt in attempts.iterator(chunk_size=500):
         workspace = attempt.workspace
 
         # Get number of blocks from solution - 1 to discount Start block