1
Fork 0

feat: simulation config

This commit is contained in:
Tibo De Peuter 2026-04-15 19:39:39 +02:00
parent 143dbbc710
commit a724716117
Signed by: tdpeuter
SSH key fingerprint: SHA256:u/h/LVoqKF1Iz02uOyxe6hcjmoZASCGV2HM0TG9ZMoU
8 changed files with 52 additions and 15 deletions

View file

@ -13,9 +13,10 @@ def test_config_composition_centralized():
# We compose the config; it follows main_config.yaml
cfg = compose(config_name="main_config", overrides=["architecture=centralized"])
# Merge with the dataclass class to get a structured DictConfig,
# then convert to a real dataclass instance to verify validation.
structured_cfg = OmegaConf.to_object(OmegaConf.merge(BrittleStarConfig, cfg))
# Merge with the structured schema and convert to a real dataclass instance to verify validation.
structured_cfg = OmegaConf.to_object(
OmegaConf.merge(OmegaConf.structured(BrittleStarConfig), cfg)
)
# Basic assertions
assert structured_cfg.architecture.name == "centralized"
@ -30,7 +31,9 @@ def test_config_composition_decentralized():
cfg = compose(config_name="main_config", overrides=["architecture=decentralized"])
# Merge and convert to dataclass instance
structured_cfg = OmegaConf.to_object(OmegaConf.merge(BrittleStarConfig, cfg))
structured_cfg = OmegaConf.to_object(
OmegaConf.merge(OmegaConf.structured(BrittleStarConfig), cfg)
)
# Basic assertions
assert structured_cfg.architecture.name == "decentralized"