code: def update_registry_for_model(self, config):
      registry.register(
          self.dataset_name + "_text_vocab_size",
          self.dataset.masked_token_processor.get_vocab_size(),
      )
  
imports: from mmf.common.registry import registry
  from mmf.datasets.builders.coco.builder import COCOBuilder
  from .masked_dataset import MaskedCOCODataset
label_desc: Use image node_id 2 for calling the MaskedCOCOBuilder obj's underlying member method code with example usage: obj.update_registry_for_model(config) without return types