fix(train): integrate YAML config

2026-03-31 23:05:43 +02:00 · 2026-03-31 23:05:43 +02:00 · e160a55d95
commit e160a55d95
parent 2001a92e75
3 changed files with 10 additions and 2 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@ -17,6 +17,7 @@ dependencies = [
    "optax>=0.2.6",
    "pyopengl>=3.1.10",
    "pyopengl-accelerate>=3.1.10",
+    "pyyaml>=6.0",
    "tyro>=1.0.10",
    "wandb==0.24.2",
 ]