Fix bug with TaskQueueSelector and blocked queues

components/scheduler/base/work_queue.h

Index: components/scheduler/base/work_queue.h
diff --git a/components/scheduler/base/work_queue.h b/components/scheduler/base/work_queue.h
index 17a7a5159df08e520d180f592045f972d7709451..23845029f3c2fe2b308529ce7106d29c094074c0 100644
--- a/components/scheduler/base/work_queue.h
+++ b/components/scheduler/base/work_queue.h
@@ -26,8 +26,10 @@ class SCHEDULER_EXPORT WorkQueue {
 
   // Associates this work queue with the given work queue sets. This must be
   // called before any tasks can be inserted into this work queue.
-  void AssignToWorkQueueSets(WorkQueueSets* work_queue_sets,
-                             size_t work_queue_set_index);
+  void AssignToWorkQueueSets(WorkQueueSets* work_queue_sets);
+
+  // Assigns the current set index.
+  void AssignSetIndex(size_t work_queue_set_index);
 
   void AsValueInto(base::trace_event::TracedValue* state) const;