Skip to content

Commit cc940d6

Browse files
dirkgrclaude
andcommitted
Fix style for ep_config addition
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent c91f523 commit cc940d6

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

src/olmo_core/internal/experiment.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,13 @@ def run(self, config: ExperimentConfig):
146146
train(config)
147147
teardown_training_environment()
148148
elif self == SubCmd.train_single:
149-
for parallelism_style in ["dp_config", "tp_config", "cp_config", "pp_config", "ep_config"]:
149+
for parallelism_style in [
150+
"dp_config",
151+
"tp_config",
152+
"cp_config",
153+
"pp_config",
154+
"ep_config",
155+
]:
150156
if (p_config := getattr(config.train_module, parallelism_style, None)) is not None:
151157
log.warning(
152158
"'%s' is set to %s, but you can't use parallelism when running on a single node. Disabling.",

0 commit comments

Comments
 (0)