code: def set_test_aug(self):
      if not self.is_test_aug:
          from easydict import EasyDict as edict
  
          cfg = edict()
          cfg.aug_modes = ["test-aug"]
          cfg.input_size = self.input_size
          cfg.task = 0
          self.transform = get_aug_transform(cfg)
          self.is_test_aug = True
  
imports: import numbers
  import os
  import os.path
  import pickle
  import queue
  import threading
  import logging
  import numbers
  import math
  import pandas
  from scipy.spatial.transform import Rotation
  import mxnet
  from pathlib import Path
  import numpy
  import torch
  from torch.utils.data import DataLoader, Dataset
  from skimage import transform
  import cv2
  import albumentations
  from albumentations.pytorch import ToTensorV2
  from augs import *
label_desc: Use image node_id 2 for calling the FaceDataset obj's underlying member method code with example usage: obj.set_test_aug() without return types