code: def labels():
      return torch.randint(10, (1000,))
  
file_path: benchmark_VAE/tests/test_datasets.py
imports: import os
  import pytest
  import torch
  from pythae.data.datasets import BaseDataset
label_desc: Use image node_id 2 for calling a global function with example usage: labels() and returns: torch