Skip to main content

entrenar/config/train/
mod.rs

1//! Single-command training from YAML configuration
2//!
3//! This module provides the main entry points for declarative training via YAML configs.
4
5#[cfg(feature = "parquet")]
6mod arrow;
7mod batches;
8mod demo;
9mod loader;
10
11// Re-export public API (allow unused for external consumers)
12#[allow(unused_imports)]
13pub use batches::{load_training_batches, rebatch};
14#[allow(unused_imports)]
15pub use demo::create_demo_batches;
16pub use loader::{load_config, train_from_yaml, try_load_apr_for_inference};
17
18// Conditionally export parquet/json loaders for non-WASM
19#[cfg(all(not(target_arch = "wasm32"), feature = "parquet"))]
20#[allow(unused_imports)]
21pub use arrow::arrow_array_to_f32;
22#[cfg(not(target_arch = "wasm32"))]
23#[allow(unused_imports)]
24pub use batches::load_json_batches;
25#[cfg(all(not(target_arch = "wasm32"), feature = "parquet"))]
26#[allow(unused_imports)]
27pub use batches::load_parquet_batches;
28
29#[cfg(test)]
30mod tests;