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

use clap::Parser;

use crate::{
    cli_commands::local::{
        inference::{self, LocalInferenceRunArgs},
        training::{self, LocalTrainingRunArgs},
    },
    context::HeatCliContext,
};

/// Run a training or inference locally.
/// Only local training is supported at the moment.
#[derive(Parser, Debug)]
pub enum LocalRunSubcommand {
    /// Run a training locally.
    Training(LocalTrainingRunArgs),
    /// Run an inference locally.
    Inference(LocalInferenceRunArgs),
}

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