entrenar 0.7.13

Training & Optimization library with autograd, LoRA, quantization, and model merging
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

//! Compilation outcome types for CITL trainer

use super::SourceSpan;
use serde::{Deserialize, Serialize};

/// Outcome of a compilation session
#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum CompilationOutcome {
    /// Compilation succeeded
    Success,
    /// Compilation failed with errors
    Failure {
        /// Error codes encountered
        error_codes: Vec<String>,
        /// Error spans
        error_spans: Vec<SourceSpan>,
        /// Error messages
        messages: Vec<String>,
    },
}

impl CompilationOutcome {
    /// Create a success outcome
    #[must_use]
    pub fn success() -> Self {
        Self::Success
    }

    /// Create a failure outcome
    #[must_use]
    pub fn failure(
        error_codes: Vec<String>,
        error_spans: Vec<SourceSpan>,
        messages: Vec<String>,
    ) -> Self {
        Self::Failure { error_codes, error_spans, messages }
    }

    /// Check if the outcome is success
    #[must_use]
    pub fn is_success(&self) -> bool {
        matches!(self, Self::Success)
    }

    /// Get error codes if failure
    #[must_use]
    pub fn error_codes(&self) -> Vec<&str> {
        match self {
            Self::Success => vec![],
            Self::Failure { error_codes, .. } => error_codes.iter().map(String::as_str).collect(),
        }
    }

    /// Get error spans if failure
    #[must_use]
    pub fn error_spans(&self) -> Vec<&SourceSpan> {
        match self {
            Self::Success => vec![],
            Self::Failure { error_spans, .. } => error_spans.iter().collect(),
        }
    }
}

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

    #[test]
    fn test_compilation_outcome_success() {
        let outcome = CompilationOutcome::success();
        assert!(outcome.is_success());
        assert!(outcome.error_codes().is_empty());
    }

    #[test]
    fn test_compilation_outcome_failure() {
        let outcome = CompilationOutcome::failure(
            vec!["E0308".to_string()],
            vec![SourceSpan::line("main.rs", 5)],
            vec!["type mismatch".to_string()],
        );
        assert!(!outcome.is_success());
        assert_eq!(outcome.error_codes(), vec!["E0308"]);
        assert_eq!(outcome.error_spans().len(), 1);
    }
}