code: def data():
      return torch.randn(1000, 2)
  
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 1 for calling a global function with example usage: data() and returns: torch