code: def _custom_collate(self, batch) -> VideoDataset:
      video, labels = torch.utils.data.default_collate(batch)
      # transform to VideoData format
      assert (
          labels == labels[0]
      ).all(), "Error: more than one video in a sample"
      batch = {"video": video, "label": labels}
      batch = torch.utils.data.default_collate([batch])
      return VideoDataset(video_tensor=batch["video"])
  
file_path: NeuralCompression/projects/torch_vct/datamodules/uvg.py
imports: from typing import Optional
  import torch
  import torch.utils.data
  import torchvision
  from datamodules.video_data_api import VideoDataset
  from pytorch_lightning import LightningDataModule
  from pytorchvideo.transforms import Normalize
  from torchvision.transforms import Compose, ToTensor
label_desc: Use image node_id 2 for calling the UVGDataModule obj's underlying member method code with example usage: obj._custom_collate(batch) and returns: VideoDataset