OmegaConf

[arturtoshev]

Replace simple yaml with OmegaConf and refactor the library.

[codecov]

Codecov Report

Attention: Patch coverage is 86.91589% with 42 lines in your changes are missing coverage. Please review.

Project coverage is 83.4%. Comparing base (ee9167a) to head (c4d698a).

Files	Patch %	Lines
lagrangebench/runner.py	79.2%	26 Missing ⚠️
lagrangebench/train/trainer.py	84.5%	11 Missing ⚠️
lagrangebench/data/data.py	80.0%	2 Missing ⚠️
lagrangebench/evaluate/rollout.py	83.3%	2 Missing ⚠️
lagrangebench/models/painn.py	50.0%	1 Missing ⚠️

Additional details and impacted files

@@           Coverage Diff            @@
##            main     #28      +/-   ##
========================================
+ Coverage   69.4%   83.4%   +13.9%     
========================================
  Files         25      26       +1     
  Lines       1662    1875     +213     
========================================
+ Hits        1155    1565     +410     
+ Misses       507     310     -197     

Flag	Coverage Δ
unittests	83.4% <86.9%> (+13.9%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[gerkone]

Small thing but don't think this is the case. The LossConfig dataclass is passed

[arturtoshev]

Is this still relevant?

[gerkone]

Will follow up on LossConfig, and some minor typing/library usability changes