mlflow-client 0.0.1

MLflow REST API client (unofficial)
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

use serde_json::Value;

use crate::data::Param;
use crate::{Error, Result};

pub(crate) fn none_if_not_exist<T, U>(
    value: Result<T>,
    f: impl FnOnce(T) -> Result<U>,
) -> Result<Option<U>> {
    match value {
        Ok(value) => Ok(Some(f(value)?)),
        Err(e) if e.is_resource_does_not_exist() => Ok(None),
        Err(e) => Err(e),
    }
}
pub(crate) fn build_params(key: &str, value: &Value, params: &mut Vec<Param>) -> Result<()> {
    match value {
        Value::Null => {}
        Value::Bool(value) => params.push(Param {
            key: key.to_string(),
            value: value.to_string(),
        }),
        Value::Number(value) => params.push(Param {
            key: key.to_string(),
            value: value.to_string(),
        }),
        Value::String(value) => params.push(Param {
            key: key.to_string(),
            value: value.to_string(),
        }),
        Value::Array(_) => {
            return Err(Error::from_message("Array not supported"));
        }
        Value::Object(m) => {
            let s = if key.is_empty() { "" } else { "." };
            for (k, v) in m {
                build_params(&format!("{key}{s}{k}"), v, params)?;
            }
        }
    }
    Ok(())
}