code: def test_build_custom_arch(
      self, model_config_with_input_dim, custom_decoder
  ):
      model = BaseAE(
          model_config_with_input_dim, decoder=custom_decoder
      )
  
      assert model.decoder == custom_decoder
      assert not model.model_config.uses_default_decoder
  
      model = BaseAE(model_config_with_input_dim)
  
      assert model.model_config.uses_default_decoder
  
      model = BaseAE(
          model_config_with_input_dim, decoder=custom_decoder
      )
  
      assert model.decoder == custom_decoder
      assert not model.model_config.uses_default_decoder
  
file_path: benchmark_VAE/tests/test_baseAE.py
imports: import os
  import shutil
  import pytest
  import torch
  from pythae.customexception import BadInheritanceError
  from pythae.models import BaseAE, BaseAEConfig
  from tests.data.custom_architectures import Decoder_AE_Conv, Encoder_AE_Conv, NetBadInheritance
label_desc: Use image node_id 3 for calling the Test_Model_Building obj's underlying member method code with example usage: obj.test_build_custom_arch(model_config_with_input_dim, custom_decoder) without return types