code: def setup_test_data(self, cfg):
      if hasattr(self, "_test_ds") and self._test_ds is not None:
          consumed_samples = 0
          logging.info(
              f"Setting up test dataloader with len(len(self._test_ds)): {len(self._test_ds)} and consumed samples: {consumed_samples}"
          )
          drop_last = True
          if not self.cfg.get("validation_drop_last", True):
              logging.info(
                  f"Drop last in validation dataset is set to False"
              )
              drop_last = False
          self._test_dl = self.build_pretraining_data_loader(
              self._test_ds, consumed_samples, drop_last
          )
  
file_path: NeMo/nemo/collections/multimodal/models/text_to_image/instruct_pix2pix/ldm/ddpm_edit.py
imports: import torch
  from einops import rearrange
  from nemo.collections.multimodal.data.instruct_pix2pix.edit_dataset import EditDataset
  from nemo.collections.multimodal.models.text_to_image.stable_diffusion.ldm.ddpm import LatentDiffusion, MegatronLatentDiffusion
  from nemo.collections.nlp.data.language_modeling.megatron.data_samplers import MegatronPretrainingRandomSampler, MegatronPretrainingSampler
  from nemo.utils import logging
label_desc: Use image node_id 6 for calling the MegatronLatentDiffusionEdit obj's underlying member method code with example usage: obj.setup_test_data(cfg) without return types