code: def __init__(self, function: Callable[[], nn.Module]):
      self.function = function
  
file_path: fairscale/fairscale/nn/pipe/types.py
imports: from dataclasses import dataclass
  from typing import Any, Callable, List, Optional, Tuple, Union
  import torch
  from torch import Tensor, nn
label_desc: Use image node_id 1 to create a new LazyModule object  with example: obj = LazyModule(function)