Remove override since we just fixed the old configs instead

mikemccabe210 · mikemccabe210 · commit aa9cb7552164 · 2025-11-20T08:42:43.000-05:00
diff --git a/walrus/train.py b/walrus/train.py
@@ -335,10 +335,6 @@ def main(cfg: DictConfig):
         cfg.distribution.distribution_type.upper() != "LOCAL" and world_size > 1
     )
 
-    # Convert cfg to a string, do find/replace from "temporary_mppx_name" to "walrus", then convert back to DictConfig
-    cfg_str = OmegaConf.to_yaml(cfg)
-    cfg_str = cfg_str.replace("temporary_mppx_name", "walrus")
-    cfg = OmegaConf.create(cfg_str)
     # Since configure_experiment uses distributed logic, distribution must be set up first
     device_mesh = configure_distribution(cfg)
     (

Original file line number	Diff line number	Diff line change
`@@ -335,10 +335,6 @@ def main(cfg: DictConfig):`
`335`	`335`	`cfg.distribution.distribution_type.upper() != "LOCAL" and world_size > 1`
`336`	`336`	`)`
`337`	`337`
`338`		`- # Convert cfg to a string, do find/replace from "temporary_mppx_name" to "walrus", then convert back to DictConfig`
`339`		`- cfg_str = OmegaConf.to_yaml(cfg)`
`340`		`- cfg_str = cfg_str.replace("temporary_mppx_name", "walrus")`
`341`		`- cfg = OmegaConf.create(cfg_str)`
`342`	`338`	`# Since configure_experiment uses distributed logic, distribution must be set up first`
`343`	`339`	`device_mesh = configure_distribution(cfg)`
`344`	`340`	`(`