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

use clap::Parser;

use crate::cli_commands::remote::{
    inference::{self, RemoteInferenceRunArgs},
    training::{self, RemoteTrainingRunArgs},
};

/// 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) -> anyhow::Result<()> {
    match args {
        RemoteRunSubcommand::Training(training_args) => training::handle_command(training_args),
        RemoteRunSubcommand::Inference(inference_args) => inference::handle_command(inference_args),
    }
}