Skip to content

Commit 4559aa9

Browse files
bgr8mrwyattii
andauthored
Refactor check_enabled root validator in DeepSpeedMonitorConfig (#3616)
* Refactor check_enabled root validator in DeepSpeedMonitorConfig * formatting * formatting --------- Co-authored-by: Michael Wyatt <[email protected]> Co-authored-by: Michael Wyatt <[email protected]>
1 parent 5d14afd commit 4559aa9

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

deepspeed/monitor/config.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,6 @@ class DeepSpeedMonitorConfig(DeepSpeedConfigModel):
7474

7575
@root_validator
7676
def check_enabled(cls, values):
77-
values["enabled"] = False
78-
if (values.get("tensorboard").enabled or values.get("wandb").enabled or values.get("csv_monitor").enabled):
79-
values["enabled"] = True
77+
values["enabled"] = values.get("tensorboard").enabled or values.get("wandb").enabled or values.get(
78+
"csv_monitor").enabled
8079
return values

0 commit comments

Comments
 (0)