euler-train 2.4.1__tar.gz → 2.5.0__tar.gz

{euler_train-2.4.1 → euler_train-2.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euler-train
-Version: 2.4.1
+Version: 2.5.0
 Summary: Lightweight file-based experiment logger for PyTorch
 Requires-Python: >=3.9
 Requires-Dist: numpy

{euler_train-2.4.1 → euler_train-2.5.0}/euler_train/outputs.py

@@ -68,10 +68,19 @@ def _save_slot(
     visualization_overrides: dict[str, Any],
 ) -> None:
     slot_dir.mkdir(parents=True, exist_ok=True)
-    items = _unpack(data)
     visualization = _resolve_visualization(
         output_type, leaf_name, visualization_overrides,
     )
+
+    # Dict with string keys → named outputs (e.g. {"scene_042": img}).
+    if isinstance(data, dict):
+        for name, raw in data.items():
+            item = _prepare(raw)
+            fmt = _resolve_format(item, output_type, leaf_name, format_overrides)
+            _save_item(slot_dir / f"{name}.{fmt}", item, fmt, visualization)
+        return
+
+    items = _unpack(data)
     for idx, item in enumerate(items):
         fmt = _resolve_format(item, output_type, leaf_name, format_overrides)
         _save_item(slot_dir / f"{idx:04d}.{fmt}", item, fmt, visualization)

{euler_train-2.4.1 → euler_train-2.5.0}/euler_train/run.py

@@ -375,6 +375,14 @@ class Run:
                             aux=dict(transmission=t_map)),
             )
 
+        Slot values can also be dicts mapping custom string IDs to
+        items, producing named files instead of sequential indices::
+
+            run.save_outputs(
+                epoch=1, step=500,
+                rgb=dict(pred={"scene_042": img_a, "scene_117": img_b}),
+            )
+
         Accepted slot keys: ``pred``, ``gt``, ``input``, ``aux``.
         ``aux`` expects a sub-dict of named arrays.
 

{euler_train-2.4.1 → euler_train-2.5.0}/euler_train.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euler-train
-Version: 2.4.1
+Version: 2.5.0
 Summary: Lightweight file-based experiment logger for PyTorch
 Requires-Python: >=3.9
 Requires-Dist: numpy

{euler_train-2.4.1 → euler_train-2.5.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "euler-train"
-version = "2.4.1"
+version = "2.5.0"
 description = "Lightweight file-based experiment logger for PyTorch"
 requires-python = ">=3.9"
 dependencies = ["numpy", "euler-dataset-contract"]

{euler_train-2.4.1 → euler_train-2.5.0}/tests/test_runlog.py

@@ -1444,6 +1444,42 @@ class TestSaveOutputsVariants:
         assert path.name == "epoch_2_step_500"
         run.finish()
 
+    def test_dict_slot_produces_named_files(self, tmp_path):
+        run = euler_train.init(dir=str(tmp_path / "r"), config={})
+        a = np.zeros((4, 4, 3), dtype=np.uint8)
+        b = np.ones((4, 4, 3), dtype=np.uint8) * 128
+        run.save_outputs(epoch=0, step=0, rgb=dict(pred={"scene_a": a, "scene_b": b}))
+
+        pred_dir = run.dir / "outputs" / "epoch_0_step_0" / "rgb" / "pred"
+        names = sorted(f.name for f in pred_dir.iterdir())
+        assert names == ["scene_a.png", "scene_b.png"]
+        run.finish()
+
+    def test_dict_slot_in_aux(self, tmp_path):
+        run = euler_train.init(dir=str(tmp_path / "r"), config={})
+        m1 = np.ones((4, 4), dtype=np.float32)
+        m2 = np.zeros((4, 4), dtype=np.float32)
+        run.save_outputs(
+            epoch=0, step=0,
+            depth=dict(aux=dict(attn={"head_0": m1, "head_1": m2})),
+        )
+
+        attn_dir = run.dir / "outputs" / "epoch_0_step_0" / "depth" / "aux" / "attn"
+        names = sorted(f.name for f in attn_dir.iterdir())
+        assert names == ["head_0.npy", "head_1.npy"]
+        run.finish()
+
+    def test_mixed_dict_and_array_slots(self, tmp_path):
+        run = euler_train.init(dir=str(tmp_path / "r"), config={})
+        a = np.zeros((4, 4, 3), dtype=np.uint8)
+        run.save_outputs(epoch=0, step=0, rgb=dict(pred={"frame_x": a}, gt=a))
+
+        pred_dir = run.dir / "outputs" / "epoch_0_step_0" / "rgb" / "pred"
+        gt_dir = run.dir / "outputs" / "epoch_0_step_0" / "rgb" / "gt"
+        assert sorted(f.name for f in pred_dir.iterdir()) == ["frame_x.png"]
+        assert sorted(f.name for f in gt_dir.iterdir()) == ["0000.png"]
+        run.finish()
+
     def test_returns_output_base_path(self, tmp_path):
         run = euler_train.init(dir=str(tmp_path / "r"), config={})
         arr = np.zeros((4, 4, 3), dtype=np.uint8)