code: def __init__(self):
      self.data = torch.randn(100, 3, 32, 32)
  
imports: import os
  import numpy
  import pytest
  import torch
  from torch import nn
  from torch.utils.data import Dataset
  from torchvision import datasets
  from torchvision.transforms import transforms
  from baal import ModelWrapper
  from baal.active.heuristics import BALD
  from baal.active.heuristics.heuristics_gpu import BALDGPUWrapper
  from baal.bayesian import Dropout
  from baal.bayesian.dropout import Dropout2d
label_desc: Use image node_id 1 to create a new SimpleDataset object from inherited base classes: Dataset with example: obj = SimpleDataset()