code: def test_caching_warnings(my_model):
      my_model = MCCachingModule(my_model)
      with warnings.catch_warnings(record=True) as tape:
          ModelWrapper(
              my_model, criterion=None, replicate_in_memory=True
          )
          assert len(tape) == 1 and "MCCachingModule" in str(
              tape[0].message
          )
  
      with warnings.catch_warnings(record=True) as tape:
          ModelWrapper(
              my_model, criterion=None, replicate_in_memory=False
          )
          assert len(tape) == 0
  
file_path: baal/tests/bayesian/test_caching.py
imports: import warnings
  import pytest
  import torch
  from torch.nn import Sequential, Linear
  from baal import ModelWrapper
  from baal.bayesian.caching_utils import MCCachingModule
label_desc: Use image node_id 3 for calling a global function with example usage: test_caching_warnings(my_model) without return types