From 9bce91969088d3229ca0e1460bc1c9207186c7dd Mon Sep 17 00:00:00 2001
From: KoalaTrapLord <265133236+KoalaTrapLord@users.noreply.github.com>
Date: Tue, 24 Mar 2026 13:45:48 -0400
Subject: [PATCH] fix: replace non-existent ForgeEpisodeGenerator import with
 actual exports

---
 ml/training/__init__.py | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/ml/training/__init__.py b/ml/training/__init__.py
index 040d41a..d26200b 100644
--- a/ml/training/__init__.py
+++ b/ml/training/__init__.py
@@ -9,8 +9,14 @@ from ml.training.rollout_buffer import RolloutBuffer
 from ml.training.reward import compute_intermediate_reward, compute_terminal_reward, RewardConfig
 from ml.training.decision_exporter import DecisionExporter
 from ml.training.forge_episode_generator import (
-    ForgeEpisodeGenerator,
-    AsyncForgeEpisodeGenerator,
+    ForgeTransition,
+    ForgeEpisode,
+    ForgeEpisodeQueue,
+    forge_producer,
+    ppo_consumer,
+    run_forge_pipeline,
+    parse_forge_jsonl,
+    snapshots_to_episode,
 )
 
 __all__ = [
@@ -23,6 +29,12 @@ __all__ = [
     # Decision export
     "DecisionExporter",
     # Forge episode generation (Phase 3)
-    "ForgeEpisodeGenerator",
-    "AsyncForgeEpisodeGenerator",
+    "ForgeTransition",
+    "ForgeEpisode",
+    "ForgeEpisodeQueue",
+    "forge_producer",
+    "ppo_consumer",
+    "run_forge_pipeline",
+    "parse_forge_jsonl",
+    "snapshots_to_episode",
 ]