atlas-cli 0.2.0

Machine Learning Lifecycle & Transparency Manager - Create and verify manifests for ML models and datasets
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

use atlas_cli::{
    cli::{
        self,
        commands::{
            CCAttestationCommands, DatasetCommands, EvaluationCommands, ManifestCommands,
            ModelCommands, PipelineCommands, SoftwareCommands,
        },
    },
    error::Result,
};
use clap::{Parser, Subcommand};

#[derive(Parser)]
#[command(author, version, about)]
pub struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    /// Dataset-related commands
    Dataset {
        #[command(subcommand)]
        command: DatasetCommands,
    },
    /// Model-related commands
    Model {
        #[command(subcommand)]
        command: ModelCommands,
    },
    /// Software component commands
    Software {
        #[command(subcommand)]
        command: SoftwareCommands,
    },
    /// Manifest management commands
    Manifest {
        #[command(subcommand)]
        command: ManifestCommands,
    },
    /// Evaluation-related commands
    Evaluation {
        #[command(subcommand)]
        command: EvaluationCommands,
    },
    /// Pipeline-related commands
    Pipeline {
        #[command(subcommand)]
        command: PipelineCommands,
    },
    /// CC Attestation-related commands
    CCAttestation {
        #[command(subcommand)]
        command: CCAttestationCommands,
    },
}

fn main() -> Result<()> {
    // Initialize logging
    atlas_cli::init_logging()?;

    // Parse command line arguments
    let cli = Cli::parse();

    // Handle commands
    let result = match cli.command {
        Commands::Dataset { command } => cli::handlers::handle_dataset_command(command),
        Commands::Model { command } => cli::handlers::handle_model_command(command),
        Commands::Software { command } => cli::handlers::handle_software_command(command),

        Commands::Manifest { command } => cli::handlers::handle_manifest_command(command),
        Commands::Evaluation { command } => cli::handlers::handle_evaluation_command(command),
        Commands::Pipeline { command } => cli::handlers::handle_pipeline_command(command),
        Commands::CCAttestation { command } => {
            cli::handlers::handle_cc_attestation_command(command)
        }
    };

    // Format and display any errors
    if let Err(ref e) = result {
        eprintln!("{}", cli::format_error(e));
    }

    result
}