batch size issue fix.

Datura-ai · Oct 15, 2024 · dd1e42a · dd1e42a
1 parent 107a658
commit dd1e42a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/validators/weight_setter.py b/validators/weight_setter.py
@@ -275,10 +275,10 @@ def pop_synthetic_tasks_max_100_per_miner(self, synthetic_tasks):
         uid_to_task_cnt = defaultdict(int)
         for uid, synthetic_task in synthetic_tasks:
             if uid_to_task_cnt[uid] < max_query_cnt_per_miner:
-                batch_tasks.append(synthetic_task)
                 if len(batch_tasks) > batch_size:
                     remain_tasks.append((uid, synthetic_task))
                     continue
+                batch_tasks.append(synthetic_task)
                 uid_to_task_cnt[uid] += 1
                 continue
             else: