(self, cfg: Config, *, parent: Optional[Module])
| 106 | validation: Optional[CheckpointValidationType] = CheckpointValidationType.EXACT |
| 107 | |
| 108 | def __init__(self, cfg: Config, *, parent: Optional[Module]): |
| 109 | super().__init__(cfg, parent=parent) |
| 110 | cfg = self.config |
| 111 | self._builders = [] |
| 112 | |
| 113 | for i, builder_cfg in enumerate(cfg.builders): |
| 114 | self._builders.append(self._add_child(f"builder{i:03d}", builder_cfg)) |
| 115 | |
| 116 | def input_state_type(self) -> Builder.StateType: |
| 117 | return self._builders[0].input_state_type() |