code: def __init__(self, stage_manager: PipelineStageManager) -> None:
      self.stage_manager = stage_manager
  
file_path: colossalai/colossalai/pipeline/schedule/base.py
imports: from typing import Any, Callable, Iterable, Optional
  from torch import Tensor
  from torch.nn import Module
  from colossalai.interface import OptimizerWrapper
  from colossalai.pipeline.stage_manager import PipelineStageManager
label_desc: Use image node_id 1 to create a new PipelineSchedule object  with example: obj = PipelineSchedule(stage_manager)