auto-sync 2026-07-03T15:40:37Z workspace (part 5)

workspace/scripts/train_ctt.py

@@ -236,7 +236,9 @@ def load_charts(
             metadata_values = data["metadata_json"]
             for row in range(chart_ids.shape[0]):
                 chart_id = str(chart_ids[row])
-                metadata = _json_loads(str(metadata_values[row]))
+                metadata = _json_loads(str(metadata_values[row])) | {
+                    "_chart_root": str(index_path.parent)
+                }
                 item = grouped.setdefault(
                     chart_id,
                     {

workspace/tests/test_chart_features.py

@@ -3,7 +3,12 @@ from __future__ import annotations
 import numpy as np
 import pytest
 
-from cil.chart_features import CONTEXT_HASH_WIDTH, build_chart_feature, chart_feature_dim
+from cil.chart_features import (
+    CONTEXT_HASH_WIDTH,
+    OBSERVATION_EMBED_DIM,
+    build_chart_feature,
+    chart_feature_dim,
+)
 
 
 def test_base_chart_feature_preserves_original_flattened_base_action() -> None:
@@ -54,6 +59,31 @@ def test_base_context_feature_does_not_read_outcome_or_hidden_fields() -> None:
     )
 
 
+def test_base_context_obs_appends_precomputed_observation_embedding(tmp_path) -> None:
+    base_action = np.zeros((2, 7), dtype=np.float32)
+    embeddings = np.arange(OBSERVATION_EMBED_DIM * 2, dtype=np.float32).reshape(
+        2, OBSERVATION_EMBED_DIM
+    )
+    path = tmp_path / "obs_embeddings.npz"
+    np.savez_compressed(path, embeddings=embeddings)
+    metadata = {
+        "task_id": "PullCube-v1",
+        "instruction": "pull the cube",
+        "observation_embedding_path": f"{path.name}#embeddings/1",
+        "_chart_root": str(tmp_path),
+        "scalar_utility": -999.0,
+        "label": "negative",
+    }
+
+    feature = build_chart_feature(base_action, metadata, mode="base_context_obs")
+
+    assert feature.shape == (
+        base_action.size + 2 * CONTEXT_HASH_WIDTH + 2 + OBSERVATION_EMBED_DIM,
+    )
+    assert np.allclose(feature[-OBSERVATION_EMBED_DIM:], embeddings[1])
+    assert chart_feature_dim(base_action, mode="base_context_obs") == feature.shape[0]
+
+
 def test_unknown_chart_feature_mode_raises() -> None:
     with pytest.raises(ValueError, match="unknown chart feature mode"):
         build_chart_feature(np.zeros((1, 2), dtype=np.float32), mode="outcome")