java.util.List<org.apache.flink.runtime.state.KeyGroupsStateHandle> to org.apache.flink.runtime.state.TaskStateHandles