def _check_masked_lm(self) -> None:
    if (self.component_config[MASKED_LM] and (self.component_config[NUM_TRANSFORMER_LAYERS] == 0)):
        raise ValueError(f"If number of transformer layers is 0, '{MASKED_LM}' option should be 'False'.")
def _check_masked_lm(self) -> None:
    if (self.component_config[MASKED_LM] and (self.component_config[NUM_TRANSFORMER_LAYERS] == 0)):
        raise ValueError(f"If number of transformer layers is 0, '{MASKED_LM}' option should be 'False'.")