def __init__()

in dualpipe/dualpipe.py [0:0]

• 6 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(
        self,
        modules: Tuple[nn.Module, nn.Module],
        batch_dim: int = 0,
        process_group: Optional[dist.ProcessGroup] = None,
        rank_mapping: Optional[List[int]] = None,