brainjepa 0.2.7

Brain-JEPA fMRI Foundation Model — inference in Rust (RLX)
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

/// Typed errors for brainjepa.
///
/// All public APIs return `Result<T, BrainJepaError>` (or `anyhow::Result`
/// for the CLI).  Match on the variant to handle specific failure modes.
use std::path::PathBuf;

#[derive(Debug, thiserror::Error)]
pub enum BrainJepaError {
    // ── Config ───────────────────────────────────────────────────────────────
    #[error("unknown model variant \"{name}\" (expected vit_small, vit_base, or vit_large)")]
    UnknownVariant { name: String },

    #[error("invalid positional embedding mode \"{mode}\" (expected \"mapping\" or \"origin\")")]
    InvalidPosMode { mode: String },

    // ── Files ────────────────────────────────────────────────────────────────
    #[error("{kind} file not found: {path}")]
    FileNotFound { kind: &'static str, path: PathBuf },

    // ── Data ─────────────────────────────────────────────────────────────────
    #[error("CSV is empty or contains no valid rows: {path}")]
    EmptyCsv { path: PathBuf },

    #[error("CSV row {row} has {got} columns, expected {expected} (file: {path})")]
    InconsistentCsvRow {
        path: PathBuf,
        row: usize,
        expected: usize,
        got: usize,
    },

    #[error("gradient CSV has {got} ROIs, expected {expected}")]
    GradientRoiMismatch { expected: usize, got: usize },

    #[error("cannot downsample {src} frames to {dst} (target must be <= source)")]
    DownsampleUpscale { src: usize, dst: usize },

    // ── Weights ──────────────────────────────────────────────────────────────
    #[error("weight key not found: {key}")]
    WeightKeyMissing { key: String },

    // ── Tensor ───────────────────────────────────────────────────────────────
    #[error("tensor conversion failed: {reason}")]
    TensorConversion { reason: String },

    // ── Pass-through ─────────────────────────────────────────────────────────
    #[error(transparent)]
    Io(#[from] std::io::Error),

    #[error(transparent)]
    Other(#[from] anyhow::Error),
}

/// Convenience alias used throughout the crate.
pub type Result<T> = std::result::Result<T, BrainJepaError>;