code: def _taskaware_classification_subset(
      dataset: SupportedDataset,
      indices: Optional[Sequence[int]] = None,
      *,
      class_mapping: Optional[Sequence[int]] = None,
      transform: Optional[XTransform] = None,
      target_transform: Optional[YTransform] = None,
      transform_groups: Optional[
          Mapping[str, Tuple[XTransform, YTransform]]
      ] = None,
      initial_transform_group: Optional[str] = None,
      task_labels: Optional[Union[int, Sequence[int]]] = None,
      targets: Optional[Sequence[TTargetType]] = None,
      collate_fn: Optional[Callable[[List], Any]] = None,
  ) -> TaskAwareClassificationDataset:
      ...
  
imports: from functools import partial
  import torch
  from torch.utils.data.dataset import Subset, ConcatDataset, TensorDataset
  from avalanche.benchmarks.utils.utils import TaskSet, _count_unique, find_common_transforms_group, _init_task_labels, _init_transform_groups, _split_user_def_targets, _split_user_def_task_label, _traverse_supported_dataset
  from avalanche.benchmarks.utils.data import AvalancheDataset
  from avalanche.benchmarks.utils.transform_groups import TransformGroupDef, DefaultTransformGroups, XTransform, YTransform
  from avalanche.benchmarks.utils.data_attribute import DataAttribute
  from avalanche.benchmarks.utils.dataset_utils import SubSequence
  from avalanche.benchmarks.utils.flat_data import ConstantSequence
  from avalanche.benchmarks.utils.dataset_definitions import ISupportedClassificationDataset, ITensorDataset, IDatasetWithTargets
  from typing import List, Any, Sequence, Union, Optional, TypeVar, Callable, Dict, Tuple, Mapping, overload
label_desc: Use image node_id 9 for calling a global function with example usage: _taskaware_classification_subset(dataset, indices) without return types