code: def a_model_deep():
      return nn.Sequential(
          nn.Linear(32, 32),
          nn.Sequential(
              nn.Linear(32, 3),
              nn.ReLU(),
              nn.Linear(10, 3),
              nn.ReLU(),
              nn.Linear(3, 3),
          ),
      )
  
file_path: baal/tests/bayesian/common_test.py
imports: import pytest
  from torch import nn
  from baal.bayesian.common import replace_layers_in_module
label_desc: Use image node_id 1 for calling a global function with example usage: a_model_deep() and returns: nn