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]>
microsoft · Jun 2, 2023 · 4559aa9 · 4559aa9
1 parent 5d14afd
commit 4559aa9
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