code: def test_build_model(self, model_configs):
      model = PixelCNN(model_configs)
      assert all(
          [
              model.input_dim == np.prod(model_configs.input_dim),
          ]
      )
  
      with pytest.raises(ValidationError):
          conf = PixelCNNConfig(kernel_size=2)
  
imports: import os
  from copy import deepcopy
  import numpy
  import pytest
  import torch
  from pydantic import ValidationError
  from pythae.models import AutoModel
  from pythae.models.base.base_utils import ModelOutput
  from pythae.models.normalizing_flows import PixelCNN, PixelCNNConfig
  from pythae.pipelines import TrainingPipeline
  from pythae.trainers import BaseTrainer, BaseTrainerConfig
label_desc: Use image node_id 1 for calling the Test_Model_Building obj's underlying member method code with example usage: obj.test_build_model(model_configs) without return types