From 81ae71c88eb83dd418d0d28348b5ed2d346b0ab9 Mon Sep 17 00:00:00 2001
From: jgrusewski <jgrusewski@localhost>
Date: Mon, 23 Feb 2026 13:40:50 +0100
Subject: [PATCH] feat(ml): export data_pipeline and asset_selection types in
 prelude

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 ml/src/asset_selection/mod.rs | 3 +++
 ml/src/data_pipeline/mod.rs   | 4 ++++
 ml/src/lib.rs                 | 6 ++++++
 3 files changed, 13 insertions(+)

diff --git a/ml/src/asset_selection/mod.rs b/ml/src/asset_selection/mod.rs
index 0ebb3c352..d3438bd4a 100644
--- a/ml/src/asset_selection/mod.rs
+++ b/ml/src/asset_selection/mod.rs
@@ -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 {
diff --git a/ml/src/data_pipeline/mod.rs b/ml/src/data_pipeline/mod.rs
index 09468c066..78fbcdd3f 100644
--- a/ml/src/data_pipeline/mod.rs
+++ b/ml/src/data_pipeline/mod.rs
@@ -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};
 
diff --git a/ml/src/lib.rs b/ml/src/lib.rs
index b569b98c1..08948b663 100644
--- a/ml/src/lib.rs
+++ b/ml/src/lib.rs
@@ -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};
 }