mog 0.0.0

A commad-line tool to log ML experiments
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
87
88
89
90
91
92
93
94

use crate::env;
use crate::util;

#[derive(Debug, structopt::StructOpt)]
#[structopt(rename_all = "kebab-case")]
pub enum GetOpt {
    Artifacts(GetArtifactsOpt),
    Contexts(GetContextsOpt),
    Events(GetEventsOpt),
    Executions(GetExecutionsOpt),
}

impl GetOpt {
    pub async fn execute(&self) -> anyhow::Result<()> {
        match self {
            Self::Artifacts(opt) => opt.execute().await,
            Self::Contexts(opt) => opt.execute().await,
            Self::Events(opt) => opt.execute().await,
            Self::Executions(opt) => opt.execute().await,
        }
    }
}

#[derive(Debug, structopt::StructOpt)]
#[structopt(rename_all = "kebab-case")]
pub struct GetArtifactsOpt {
    #[structopt(long, env = env::KEY_DATABASE, hide_env_values = true)]
    pub database: String,
}

impl GetArtifactsOpt {
    pub async fn execute(&self) -> anyhow::Result<()> {
        let mut store = util::mlmd_connect(&self.database).await?;
        let artifacts = store.get_artifacts().execute().await?;
        for artifact in artifacts {
            println!("{:?}", artifact);
        }
        Ok(())
    }
}

#[derive(Debug, structopt::StructOpt)]
#[structopt(rename_all = "kebab-case")]
pub struct GetContextsOpt {
    #[structopt(long, env = env::KEY_DATABASE, hide_env_values = true)]
    pub database: String,
}

impl GetContextsOpt {
    pub async fn execute(&self) -> anyhow::Result<()> {
        let mut store = util::mlmd_connect(&self.database).await?;
        let contexts = store.get_contexts().execute().await?;
        for context in contexts {
            println!("{:?}", context);
        }
        Ok(())
    }
}

#[derive(Debug, structopt::StructOpt)]
#[structopt(rename_all = "kebab-case")]
pub struct GetEventsOpt {
    #[structopt(long, env = env::KEY_DATABASE, hide_env_values = true)]
    pub database: String,
}

impl GetEventsOpt {
    pub async fn execute(&self) -> anyhow::Result<()> {
        let mut store = util::mlmd_connect(&self.database).await?;
        let events = store.get_events().execute().await?;
        for event in events {
            println!("{:?}", event);
        }
        Ok(())
    }
}

#[derive(Debug, structopt::StructOpt)]
#[structopt(rename_all = "kebab-case")]
pub struct GetExecutionsOpt {
    #[structopt(long, env = env::KEY_DATABASE, hide_env_values = true)]
    pub database: String,
}

impl GetExecutionsOpt {
    pub async fn execute(&self) -> anyhow::Result<()> {
        let mut store = util::mlmd_connect(&self.database).await?;
        let executions = store.get_executions().execute().await?;
        for execution in executions {
            println!("{:?}", execution);
        }
        Ok(())
    }
}