code: def my_model():
      return Sequential(
          LinearMocked(10, 10),
          LinearMocked(10, 10),
          Sequential(
              LinearMocked(10, 10),
              LinearMocked(10, 10),
          ),
      ).eval()
  
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 1 for calling a global function with example usage: my_model() and returns: Sequential