text: # Copyright (c) Facebook, Inc. and its affiliates.
  # All rights reserved.
  #
  # This source code is licensed under the license found in the
  # LICENSE file in the root directory of this source tree.
  #
  
  from VisualBERT.mmf.common.registry import registry
  from VisualBERT.mmf.datasets.builders.mmimdb.masked_dataset import MaskedMMImdbDataset
  from VisualBERT.mmf.datasets.builders.vqa2.builder import VQA2Builder
  
  
  @registry.register_builder("masked_mmimdb")
  class MaskedMMImdbBuilder(VQA2Builder):
      def __init__(self):
          super().__init__()
          self.dataset_name = "masked_mmimdb"
          self.dataset_class = MaskedMMImdbDataset
  
      @classmethod
      def config_path(cls):
          return "configs/datasets/mmimdb/masked.yaml"
  