code: def create_dataloader(dataset: Dataset):
      return DataLoader(dataset=dataset, batch_size=10)
  
file_path: nni/test/algo/compression/distillation/test_distil_label_patcher.py
imports: from __future__ import annotations
  import hashlib
  import pytest
  import torch
  from torch.utils.data import Dataset, DataLoader
  from nni.contrib.distillation.label_patcher import DistilLabelPatcher
  from nni.contrib.distillation.uid_dataset import IndexedDataset, HashedDataset, AugmentationDataset
  from ..assets.common import log_dir
label_desc: Use image node_id 2 for calling a global function with example usage: create_dataloader(dataset) and returns: DataLoader