code: def get_mbatch(data, batch_size=5):
      dl = DataLoader(
          data,
          shuffle=False,
          batch_size=batch_size,
          collate_fn=data.collate_fn,
      )
      return next(iter(dl))
  
imports: import unittest
  import torch
  from numpy.testing import assert_almost_equal
  from torch.utils.data import TensorDataset, DataLoader
  from avalanche.benchmarks import benchmark_from_datasets, benchmark_with_validation_stream, CLScenario, CLStream, split_validation_random, task_incremental_benchmark
  from avalanche.benchmarks.scenarios.dataset_scenario import DatasetExperience, split_validation_class_balanced
  from avalanche.benchmarks.utils import AvalancheDataset
  from tests.unit_tests_utils import dummy_tensor_dataset, get_fast_benchmark, DummyImageDataset
label_desc: Use image node_id 1 for calling a global function with example usage: get_mbatch(data, batch_size) and returns: next