# Source code for archai.algos.xnas.xnas_exp_runner

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

    from typing import Type

    from overrides import overrides

    from archai.nas.exp_runner import ExperimentRunner
    from archai.nas.arch_trainer import TArchTrainer
    from .xnas_model_desc_builder import XnasModelDescBuilder
    from .xnas_arch_trainer import XnasArchTrainer

    class XnasExperimentRunner(ExperimentRunner):
        @overrides
        def model_desc_builder(self)->XnasModelDescBuilder:
            return XnasModelDescBuilder()

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