code: def custom_encoder(self, ae_config):
      return Encoder_AE_MLP_Custom(ae_config)
  
imports: import os
  from copy import deepcopy
  import pytest
  import torch
  from pythae.customexception import ModelError
  from pythae.models import AE, RHVAE, VAE, AEConfig, RHVAEConfig, VAEConfig
  from pythae.trainers import BaseTrainer, BaseTrainerConfig
  from tests.data.custom_architectures import *
label_desc: Use image node_id 4 for calling the Test_Device_Checks obj's underlying member method code with example usage: obj.custom_encoder(ae_config) and returns: Encoder_AE_MLP_Custom