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
28
29
30
31
32
33

pub mod inference;
pub mod training;

use clap::Parser;

use crate::{
    cli_commands::run::remote::{
        inference::RemoteInferenceRunArgs, training::RemoteTrainingRunArgs,
    },
    context::HeatCliContext,
};

/// Run a training or inference remotely.
/// Not yet supported.
#[derive(Parser, Debug)]
pub enum RemoteRunSubcommand {
    /// todo
    Training(RemoteTrainingRunArgs),
    /// todo
    Inference(RemoteInferenceRunArgs),
}

pub(crate) fn handle_command(
    args: RemoteRunSubcommand,
    context: HeatCliContext,
) -> anyhow::Result<()> {
    match args {
        RemoteRunSubcommand::Training(training_args) => {
            training::handle_command(training_args, context)
        }
        RemoteRunSubcommand::Inference(inference_args) => inference::handle_command(inference_args),
    }
}