entrenar 0.7.9

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

//! Research init subcommand

use crate::cli::logging::log;
use crate::cli::LogLevel;
use crate::config::{ArtifactTypeArg, LicenseArg, ResearchInitArgs};
use crate::research::{Affiliation, ArtifactType, Author, License, ResearchArtifact};

/// Convert CLI artifact type argument to research domain type.
fn convert_artifact_type(arg: &ArtifactTypeArg) -> ArtifactType {
    match arg {
        ArtifactTypeArg::Dataset => ArtifactType::Dataset,
        ArtifactTypeArg::Paper => ArtifactType::Paper,
        ArtifactTypeArg::Model => ArtifactType::Model,
        ArtifactTypeArg::Code => ArtifactType::Code,
        ArtifactTypeArg::Notebook => ArtifactType::Notebook,
        ArtifactTypeArg::Workflow => ArtifactType::Workflow,
    }
}

/// Convert CLI license argument to research domain license.
fn convert_license(arg: &LicenseArg) -> License {
    match arg {
        LicenseArg::CcBy4 => License::CcBy4,
        LicenseArg::CcBySa4 => License::Custom("CC-BY-SA-4.0".to_string()),
        LicenseArg::Cc0 => License::Cc0,
        LicenseArg::Mit => License::Mit,
        LicenseArg::Apache2 => License::Apache2,
        LicenseArg::Gpl3 => License::Gpl3,
        LicenseArg::Bsd3 => License::Bsd3,
    }
}

/// Build an Author from CLI arguments (name, optional ORCID and affiliation).
fn build_author(args: &ResearchInitArgs) -> Result<Option<Author>, String> {
    let Some(author_name) = &args.author else {
        return Ok(None);
    };
    let mut author = Author::new(author_name);

    if let Some(orcid) = &args.orcid {
        author = author.with_orcid(orcid).map_err(|e| format!("Invalid ORCID: {e}"))?;
    }
    if let Some(affiliation) = &args.affiliation {
        author = author.with_affiliation(Affiliation::new(affiliation));
    }
    Ok(Some(author))
}

pub fn run_research_init(args: ResearchInitArgs, level: LogLevel) -> Result<(), String> {
    log(level, LogLevel::Normal, &format!("Initializing research artifact: {}", args.id));

    let artifact_type = convert_artifact_type(&args.artifact_type);
    let license = convert_license(&args.license);
    let mut artifact = ResearchArtifact::new(&args.id, &args.title, artifact_type, license);

    if let Some(author) = build_author(&args)? {
        artifact = artifact.with_author(author);
    }

    if let Some(description) = &args.description {
        artifact = artifact.with_description(description);
    }
    if let Some(keywords) = &args.keywords {
        let kw: Vec<&str> = keywords.split(',').map(str::trim).collect();
        artifact = artifact.with_keywords(kw);
    }
    if let Some(doi) = &args.doi {
        artifact = artifact.with_doi(doi);
    }

    let yaml = serde_yaml::to_string(&artifact).map_err(|e| format!("Serialization error: {e}"))?;
    std::fs::write(&args.output, &yaml).map_err(|e| format!("Failed to write file: {e}"))?;

    log(level, LogLevel::Normal, &format!("Artifact saved to: {}", args.output.display()));

    Ok(())
}