Skip to content

Features/preview and logging#286

Open
EladMichael wants to merge 5 commits into
pnnl:developfrom
EladMichael:features/preview_and_logging
Open

Features/preview and logging#286
EladMichael wants to merge 5 commits into
pnnl:developfrom
EladMichael:features/preview_and_logging

Conversation

@EladMichael
Copy link
Copy Markdown
Collaborator

System (system.py)

  • Node: new input_map param — per-key temporal window config (past, future, pad_mode, fill)
  • SystemPreview: full redesign — preview config moves from class level to each Node.input_map; adds start_iter; supports past + future windows; 4 padding modes
  • Bug fix: _check_unique_names() never actually checked uniqueness; also ran before auto-naming, missing collisions between named and unnamed nodes — both fixed

Loggers (loggers.py)

  • MLFlowLogger: nested param for parent/child runs; save_weights flag; verbosity=0 silences stdout; missing step= arg added to all log_metric calls (bug fix); new log_dict method

Trainer (trainer.py)

  • Respects logger.save_weights before saving model artifacts

Tests (test_system.py)

  • 23 new tests: TestNodeInputMap, TestSystemPreviewGetMappedData (all pad modes + error cases), and 4 forward integration tests for SystemPreview
  • All system and logger tests passing

Examples

  • Both Part_3 and Part_6 migrated to new Node.input_map API; preview horizons decoupled from nsteps (10 and 5 steps respectively, vs 30 and 20 before)

…dded dictionary logging capability, and capability to turn off stdout print with verbosity=0 to declutter when automatic hyperparameter tuning. Added save_weights flag, again for hyperparameter tuning situations, as well as storing run_id for easy access to run parameters downstream. Minimal fix, should not break anything that already uses (or doesn't use) this logger.
…rmation, nodes control their own index mapping as well as their own padding methods and their own preview window sizes, and fixed preview length going stale after nsteps is changed.
@EladMichael EladMichael requested a review from drgona May 8, 2026 16:33
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant