feat(ml): export data_pipeline and asset_selection types in prelude

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
jgrusewski
2026-02-23 13:40:50 +01:00
parent 1cc5a79365
commit 81ae71c88e
3 changed files with 13 additions and 0 deletions

View File

@@ -8,6 +8,9 @@ use serde::{Deserialize, Serialize};
pub mod scorer;
pub mod selector;
pub use scorer::{AssetScore, PredictabilityScorer, PredictionResult, ScoringWeights};
pub use selector::{ActiveSetConfig, ActiveSetSelector, TradingCandidate};
/// Classification of a tradable asset.
#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
pub enum AssetClass {

View File

@@ -8,6 +8,10 @@ pub mod cache;
pub mod manager;
pub mod prepared;
pub use cache::{CacheManifest, CachedRange, SymbolCache};
pub use manager::{DataSplit, DatasetManager, FEATURE_DIM};
pub use prepared::{Batch, BatchIterator, PreparedDataset};
use chrono::NaiveDate;
use serde::{Deserialize, Serialize};

View File

@@ -1962,4 +1962,10 @@ pub mod prelude {
// Common external types
pub use rust_decimal::Decimal;
pub use serde::{Deserialize, Serialize};
// Data pipeline
pub use crate::data_pipeline::{DatasetManager, DatasetMode, DatasetSpec, PreparedDataset};
// Asset selection
pub use crate::asset_selection::{ActiveSetSelector, AssetUniverse, PredictabilityScorer};
}