code: protected TaskAssignor<UUID, TaskId> createTaskAssignor(Map<TaskId, Set<TopicPartition>> partitionsForTask, Map<UUID, ClientState> states, Map<UUID, OverridableStreamsPartitionAssignor.ClientMetadata> clientsMetadata) {
      return new StickyTaskAssignor<>(states, partitionsForTask.keySet());
    }
docs: TWITTER CHANGED