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

//! Device error messages — compile-time feature vs runtime availability.

#![cfg(feature = "rlx")]

use brainjepa::rlx::{ensure_device, parse_device};

#[test]
fn mlx_without_feature_gives_build_hint() {
    if cfg!(feature = "rlx-mlx") {
        return;
    }
    let err = ensure_device(rlx::Device::Mlx).unwrap_err().to_string();
    assert!(err.contains("rlx-mlx"), "expected feature hint: {err}");
    assert!(
        err.contains("cargo build"),
        "expected mlx rebuild hint: {err}"
    );
}

#[test]
fn metal_without_feature_gives_build_hint() {
    if cfg!(feature = "rlx-metal") {
        return;
    }
    let err = ensure_device(rlx::Device::Metal).unwrap_err().to_string();
    assert!(err.contains("rlx-metal"), "expected feature hint: {err}");
    assert!(
        err.contains("cargo build"),
        "expected rebuild command: {err}"
    );
}

#[test]
fn parse_wgpu_alias() {
    assert_eq!(parse_device("wgpu").unwrap(), rlx::Device::Gpu);
}