code: def setup(self, stage: str) -> None:
      print("Preparing datasets")
      if self.dummy:
          data_fn = create_dummy_dataset
      else:
          data_fn = create_dataset
  
      self.train_datapipeline = data_fn(
          **self.train_config.datapipeline
      )
      if self.val_config:
          self.val_datapipeline = data_fn(
              **self.val_config.datapipeline
          )
      if self.test_config:
          self.test_datapipeline = data_fn(
              **self.test_config.datapipeline
          )
  
file_path: generative-models/sgm/data/dataset.py
imports: from typing import Optional
  import torchdata.datapipes.iter
  import webdataset
  from omegaconf import DictConfig
  from pytorch_lightning import LightningDataModule
label_desc: Use image node_id 2 for calling the StableDataModuleFromConfig obj's underlying member method code with example usage: obj.setup(stage) without return types