polypixel-memoir-core 0.4.0

Memoir memory substrate as an embeddable Rust library
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

/// Failure modes for the [`crate::nli::NliClassifier`].
#[derive(Debug, thiserror::Error)]
pub enum NliError {
    /// The model or tokenizer file could not be downloaded from HuggingFace.
    #[error("failed to download NLI model: {0}")]
    Download(String),

    /// The ONNX model could not be loaded into an inference session.
    #[error("failed to load NLI model: {0}")]
    ModelLoad(String),

    /// The tokenizer file could not be parsed.
    #[error("failed to load NLI tokenizer: {0}")]
    TokenizerLoad(String),

    /// Tokenization or model inference failed at classification time.
    #[error("NLI inference error: {0}")]
    Inference(String),
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn should_render_model_load_error_with_message() {
        let err = NliError::ModelLoad("session init failed".to_string());

        assert_eq!(err.to_string(), "failed to load NLI model: session init failed");
    }

    #[test]
    fn should_render_inference_error_with_message() {
        let err = NliError::Inference("tokenization failed".to_string());

        assert_eq!(err.to_string(), "NLI inference error: tokenization failed");
    }
}