(self)
| 118 | self.dependencies = defaultdict(set) |
| 119 | |
| 120 | def init(self): |
| 121 | # We normalize self.dependencies to Dict[Type[Stage], Set[Type[Stage]]]. |
| 122 | dependencies = defaultdict(set) |
| 123 | for k, v in self.dependencies.items(): |
| 124 | dependencies[get_stage_type(k)] = set(get_stage_type(x) for x in v) |
| 125 | self.dependencies = dependencies |
| 126 | |
| 127 | def set_mwm_stage(self, stage: Type[Stage]): |
| 128 | assert self.mwm_stage is None |
no test coverage detected