class TaskExecutorPartitionTrackerImpl extends AbstractPartitionTracker<JobID, TaskExecutorPartitionInfo> implements TaskExecutorPartitionTracker {

	@Override
	public void stopTrackingAndReleaseAllClusterPartitions() {
		clusterPartitions.values().stream().map(DataSetEntry::getPartitionIds).forEach(shuffleEnvironment::releasePartitionsLocally);
		clusterPartitions.clear();
	}

	public  TaskExecutorPartitionTrackerImpl(ShuffleEnvironment<?, ?> shuffleEnvironment);

}