code: def __init__(self, in_features: int, out_features: int):
      super().__init__(in_features, out_features)
  
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 to create a new LinearMocked object from inherited base classes: Linear with example: obj = LinearMocked(in_features, out_features)