entrenar 0.7.11

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

//! CITL configuration types

/// Configuration for the CITL trainer
#[derive(Debug, Clone)]
pub struct CITLConfig {
    /// Maximum number of fix suggestions to return
    pub max_suggestions: usize,
    /// Minimum suspiciousness score to report
    pub min_suspiciousness: f32,
    /// Whether to build dependency graphs
    pub enable_dependency_graph: bool,
}

impl Default for CITLConfig {
    fn default() -> Self {
        Self { max_suggestions: 5, min_suspiciousness: 0.3, enable_dependency_graph: true }
    }
}

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

    #[test]
    fn test_citl_config_default() {
        let config = CITLConfig::default();
        assert_eq!(config.max_suggestions, 5);
        assert!((config.min_suspiciousness - 0.3).abs() < 0.01);
        assert!(config.enable_dependency_graph);
    }
}