code: def test_without_nonlinearity(
      output_dim, hidden_channels, kernel_sizes, strides
  ):
      input_width = 32
      input_height = 32
      in_channel = 3
      input_shape = (in_channel, input_height, input_width)
      spec = InOutSpec(
          akro.Box(shape=input_shape, low=-np.inf, high=np.inf),
          akro.Box(shape=(output_dim,), low=-np.inf, high=np.inf),
      )
  
      module = DiscreteCNNModule(
          spec=spec,
          image_format="NCHW",
          hidden_channels=hidden_channels,
          hidden_sizes=hidden_channels,
          kernel_sizes=kernel_sizes,
          strides=strides,
          mlp_hidden_nonlinearity=None,
          cnn_hidden_nonlinearity=None,
          hidden_w_init=nn.init.ones_,
          output_w_init=nn.init.ones_,
      )
  
      assert len(module._module) == 3
  
file_path: garage/tests/garage/torch/modules/test_discrete_cnn_module.py
imports: import pickle
  import akro
  import numpy
  import pytest
  import torch
  import torch.nn
  from garage import InOutSpec
  from garage.torch.modules import CNNModule, DiscreteCNNModule, MLPModule
label_desc: Use image node_id 2 for calling a global function with example usage: test_without_nonlinearity(output_dim, hidden_channels, kernel_sizes, strides) without return types