heat-sdk-cli 0.1.0

Tracel Heat SDK CLI.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

use clap::Parser;

use crate::context::HeatCliContext;

use crate::cli_commands::{
    local::{self, LocalRunSubcommand},
    remote::{self, RemoteRunSubcommand},
};

/// Run a training or inference locally or trigger a remote run.
/// Only local training is supported at the moment.
#[derive(Parser, Debug)]
pub enum RunLocationType {
    /// {training|inference} : Run a training or inference locally.
    #[command(subcommand)]
    Local(LocalRunSubcommand),
    /// todo
    #[command(subcommand)]
    Remote(RemoteRunSubcommand),
}

pub(crate) fn handle_command(args: RunLocationType, context: HeatCliContext) -> anyhow::Result<()> {
    match args {
        RunLocationType::Local(local_args) => local::handle_command(local_args, context),
        RunLocationType::Remote(remote_args) => remote::handle_command(remote_args),
    }
}