# 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 archai.algos.xnas.xnas_model_desc_builder import XnasModelDescBuilder
from archai.algos.xnas.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

