def METHOD_NAME(cls, state, **kwargs):
    return super().METHOD_NAME(
        state,
        embeddings=state.get("embeddings"),
        label_dictionary=state.get("label_dictionary"),
        label_type=state.get("label_type"),
        entity_label_type=state.get("entity_label_type"),
        loss_weights=state.get("weight_dict"),
        pooling_operation=state.get("pooling_operation"),
        entity_pair_filters=state.get("entity_pair_filters"),
        train_on_gold_pairs_only=state.get("train_on_gold_pairs_only", False),
        **kwargs,
    )