# Copyright (c) Facebook, Inc. and its affiliates.

from VisualBERT.mmf.common.registry import registry
from VisualBERT.mmf.datasets.builders.gqa.builder import GQABuilder
from VisualBERT.mmf.datasets.builders.gqa.masked_dataset import MaskedGQADataset


@registry.register_builder("masked_gqa")
class MaskedGQABuilder(GQABuilder):
    def __init__(self):
        super().__init__()
        self.dataset_name = "masked_gqa"
        self.dataset_class = MaskedGQADataset

    @classmethod
    def config_path(cls):
        return "configs/datasets/gqa/masked.yaml"
