code: def train_dataset(self):
      return torch.load(
          os.path.join(PATH, "data/mnist_clean_train_dataset_sample")
      )
  
imports: import os
  from collections import Counter
  import pytest
  import torch
  from pythae.models import AE, RAE_L2, VAEGAN, Adversarial_AE, Adversarial_AE_Config, AEConfig, RAE_L2_Config, VAEGANConfig
  from pythae.trainers import *
  from pythae.trainers.training_callbacks import *
label_desc: Use image node_id 1 for calling the Test_TrainersCalls obj's underlying member method code with example usage: obj.train_dataset() and returns: torch