Major Description
Task manager needs to call `createTasks` inside partitionsAssigned callback, which is after the `onAssignment` callback for assignor. This means during task creation we rely on the status change based on the intermediate data structures populated by a different callback, which is hard to reason about. We should consider consolidate logics to either one of the callbacks, prefer `onAssignment` as it contains full information needed.