def __init__()

in ultravox/tools/mds_tool.py [0:0]

• 14 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, args: MdsArgs):
        self._args = args
        if self._args.verbose:
            logging.basicConfig(level=logging.INFO)

        print("Loading dataset...")
        self._dataset = datasets.load_dataset(
            self._args.dataset_name,
            self._args.dataset_subset,
            split=self._args.dataset_split,
            trust_remote_code=True,
        )
        logging.info(
            f"Loaded {self._dataset}, subset={self._args.dataset_subset} split={self._args.dataset_split}"
        )