Refactor check_enabled root validator in DeepSpeedMonitorConfig (#3616)

* Refactor check_enabled root validator in DeepSpeedMonitorConfig * formatting * formatting --------- Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com> Co-authored-by: Michael Wyatt <mrwyattii@gmail.com>
microsoft · Jun 23, 2023 · 45d3b03 · 45d3b03
1 parent bb967f1
commit 45d3b03
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/deepspeed/monitor/config.py b/deepspeed/monitor/config.py
@@ -74,7 +74,6 @@ class DeepSpeedMonitorConfig(DeepSpeedConfigModel):
 
     @root_validator
     def check_enabled(cls, values):
-        values["enabled"] = False
-        if (values.get("tensorboard").enabled or values.get("wandb").enabled or values.get("csv_monitor").enabled):
-            values["enabled"] = True
+        values["enabled"] = values.get("tensorboard").enabled or values.get("wandb").enabled or values.get(
+            "csv_monitor").enabled
         return values