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

//! Google Cloud Storage configuration

use serde::{Deserialize, Serialize};

/// Google Cloud Storage configuration
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct GCSConfig {
    /// GCS bucket name
    pub bucket: String,
    /// Object prefix
    pub prefix: String,
    /// Project ID
    pub project_id: Option<String>,
    /// Service account JSON key path
    pub service_account_key: Option<String>,
}

impl GCSConfig {
    /// Create a new GCS configuration
    pub fn new(bucket: &str) -> Self {
        Self {
            bucket: bucket.to_string(),
            prefix: String::new(),
            project_id: None,
            service_account_key: None,
        }
    }

    /// Set object prefix
    pub fn with_prefix(mut self, prefix: &str) -> Self {
        self.prefix = prefix.to_string();
        self
    }

    /// Set project ID
    pub fn with_project(mut self, project_id: &str) -> Self {
        self.project_id = Some(project_id.to_string());
        self
    }
}

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

    #[test]
    fn test_gcs_config_new() {
        let config = GCSConfig::new("my-bucket");
        assert_eq!(config.bucket, "my-bucket");
    }

    #[test]
    fn test_gcs_config_with_project() {
        let config = GCSConfig::new("bucket").with_project("my-project");
        assert_eq!(config.project_id, Some("my-project".to_string()));
    }

    #[test]
    fn test_gcs_config_with_prefix() {
        let config = GCSConfig::new("bucket").with_prefix("models/v1/");
        assert_eq!(config.prefix, "models/v1/");
    }

    #[test]
    fn test_gcs_config_serde() {
        let config = GCSConfig::new("bucket").with_prefix("artifacts/").with_project("my-project");

        let json = serde_json::to_string(&config).expect("JSON serialization should succeed");
        let parsed: GCSConfig =
            serde_json::from_str(&json).expect("JSON deserialization should succeed");

        assert_eq!(config.bucket, parsed.bucket);
        assert_eq!(config.prefix, parsed.prefix);
        assert_eq!(config.project_id, parsed.project_id);
    }
}