|
| 1 | +# -------------------------------------------------------------------------------------------- |
| 2 | +# Copyright (c) Microsoft Corporation. All rights reserved. |
| 3 | +# Licensed under the MIT License. See License.txt in the project root for license information. |
| 4 | +# |
| 5 | +# Code generated by aaz-dev-tools |
| 6 | +# -------------------------------------------------------------------------------------------- |
| 7 | + |
| 8 | +from azure.cli.core import AzCommandsLoader |
| 9 | +from azext_weights_and_biases._help import helps # pylint: disable=unused-import |
| 10 | + |
| 11 | + |
| 12 | +class WeightsAndBiasesCommandsLoader(AzCommandsLoader): |
| 13 | + |
| 14 | + def __init__(self, cli_ctx=None): |
| 15 | + from azure.cli.core.commands import CliCommandType |
| 16 | + custom_command_type = CliCommandType( |
| 17 | + operations_tmpl='azext_weights_and_biases.custom#{}') |
| 18 | + super().__init__(cli_ctx=cli_ctx, |
| 19 | + custom_command_type=custom_command_type) |
| 20 | + |
| 21 | + def load_command_table(self, args): |
| 22 | + from azext_weights_and_biases.commands import load_command_table |
| 23 | + from azure.cli.core.aaz import load_aaz_command_table |
| 24 | + try: |
| 25 | + from . import aaz |
| 26 | + except ImportError: |
| 27 | + aaz = None |
| 28 | + if aaz: |
| 29 | + load_aaz_command_table( |
| 30 | + loader=self, |
| 31 | + aaz_pkg_name=aaz.__name__, |
| 32 | + args=args |
| 33 | + ) |
| 34 | + load_command_table(self, args) |
| 35 | + return self.command_table |
| 36 | + |
| 37 | + def load_arguments(self, command): |
| 38 | + from azext_weights_and_biases._params import load_arguments |
| 39 | + load_arguments(self, command) |
| 40 | + |
| 41 | + |
| 42 | +COMMAND_LOADER_CLS = WeightsAndBiasesCommandsLoader |
0 commit comments