fix: preserve training venv in cpu smoke

Author: Shaw

.github/workflows/training-stack.yml

@@ -75,10 +75,12 @@ jobs:
       - name: Import probe — model_registry
         # The whole point of this lane: catch import-time errors in the
         # training package before they hit a real GPU run.
+        # Mount live sources away from /workspace so the image's
+        # /workspace/.venv remains visible at runtime.
         run: |
           docker run --rm \
-            -v "${{ github.workspace }}/packages/training:/workspace" \
-            -e PYTHONPATH=/workspace/scripts \
+            -v "${{ github.workspace }}/packages/training:/workspace_src" \
+            -e PYTHONPATH=/workspace_src/scripts \
             eliza-training-cpu:ci \
             -c "python -c 'from training.model_registry import REGISTRY; print(list(REGISTRY))'"
 
@@ -87,11 +89,11 @@ jobs:
         # may be a no-op in some branches; we still surface its result.
         run: |
           docker run --rm \
-            -v "${{ github.workspace }}/packages/training:/workspace" \
-            -e PYTHONPATH=/workspace/scripts \
+            -v "${{ github.workspace }}/packages/training:/workspace_src" \
+            -e PYTHONPATH=/workspace_src/scripts \
             eliza-training-cpu:ci \
             -c "set -e; \
-                cd /workspace; \
+                cd /workspace_src; \
                 pytest -xvs \
                   scripts/test_default_thought_leak.py \
                   scripts/test_eliza_record.py \
@@ -106,10 +108,10 @@ jobs:
         # COVERAGE_AUDIT.md) on every PR.
         run: |
           docker run --rm \
-            -v "${{ github.workspace }}/packages/training:/workspace" \
-            -e PYTHONPATH=/workspace/scripts \
+            -v "${{ github.workspace }}/packages/training:/workspace_src" \
+            -e PYTHONPATH=/workspace_src/scripts \
             eliza-training-cpu:ci \
-            -c "set -e; cd /workspace; bash scripts/ci_smoke_e2e.sh"
+            -c "set -e; cd /workspace_src; bash scripts/ci_smoke_e2e.sh"
 
       - name: Upload logs on failure
         if: failure()

packages/training/scripts/training/__init__.py

@@ -5,16 +5,29 @@
 training pipeline (APOLLO, APOLLO-Mini, AdamW baseline).
 """
 
-from .optimizer import (
-    build_adamw_optimizer,
-    build_apollo_mini_optimizer,
-    build_apollo_optimizer,
-    optimizer_state_bytes,
-)
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from .optimizer import (
+        build_adamw_optimizer,
+        build_apollo_mini_optimizer,
+        build_apollo_optimizer,
+        optimizer_state_bytes,
+    )
 
 __all__ = [
     "build_adamw_optimizer",
     "build_apollo_mini_optimizer",
     "build_apollo_optimizer",
     "optimizer_state_bytes",
 ]
+
+
+def __getattr__(name: str) -> Any:
+    if name in __all__:
+        from . import optimizer
+
+        return getattr(optimizer, name)
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")