# Source code for archai.algos.darts.darts_exp_runner

    # Copyright (c) Microsoft Corporation.
    # Licensed under the MIT license.

    from typing import Optional

    from overrides import overrides

    from archai.nas.exp_runner import ExperimentRunner
    from archai.nas.arch_trainer import TArchTrainer
    from archai.algos.darts.darts_model_desc_builder import DartsModelDescBuilder
    from archai.algos.darts.bilevel_arch_trainer import BilevelArchTrainer

    class DartsExperimentRunner(ExperimentRunner):
        @overrides
        def model_desc_builder(self)->DartsModelDescBuilder:
            return DartsModelDescBuilder()

        @overrides
        def trainer_class(self)->TArchTrainer:
            return BilevelArchTrainer