code: def test_replace_layers_in_module_swap_no_relu_deep(a_model_deep):
      mapping = (
          lambda mod: None
          if not isinstance(mod, nn.ReLU6)
          else nn.Identity()
      )
      changed = replace_layers_in_module(a_model_deep, mapping)
      assert not changed
      assert any(isinstance(m, nn.ReLU) for m in a_model_deep.modules())
      assert not any(
          isinstance(m, nn.Identity) for m in a_model_deep.modules()
      )
  
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 5 for calling a global function with example usage: test_replace_layers_in_module_swap_no_relu_deep(a_model_deep) without return types