code: def __update_model_class_inheritance(cls):
      """
      Updates the class inheritance of a torch.nn.Module to instead use
      crypten.nn.Module
      """
      bases = []
      for m in cls.__bases__:
          if m == torch.nn.Module:
              bases.append(cnn.Module)
          elif m == torch.nn.Sequential:
              bases.append(cnn.Sequential)
          elif m == torch.nn.ModuleDict:
              bases.append(cnn.ModuleDict)
          else:
              bases.append(m)
  
      cls.__bases__ = tuple(bases)
  
file_path: CrypTen/crypten/models/__init__.py
imports: import importlib.util
  import logging
  import sys
  import crypten.nn
  import torch
label_desc: Use image node_id 3 for calling a global function with example usage: __update_model_class_inheritance(cls) without return types