Format

Author: rahul003

src/sagemaker/fw_utils.py

@@ -142,7 +142,14 @@ def validate_bool(keys):
     for key in ["microbatches", "partitions", "active_microbatches"]:
         validate_positive(key)
 
-    for key in ["auto_partition", "contiguous", "load_partition", "horovod", "ddp", "deterministic_server"]:
+    for key in [
+        "auto_partition",
+        "contiguous",
+        "load_partition",
+        "horovod",
+        "ddp",
+        "deterministic_server",
+    ]:
         validate_bool(key)
 
     if "partition_file" in config and not isinstance(config.get("partition_file"), str):

tests/unit/test_fw_utils.py

@@ -115,7 +115,7 @@ def test_mp_config_string_names(pipeline, placement_strategy, optimize, trace_de
         "optimize": optimize,
         "trace_device": trace_device,
         "active_microbatches": 8,
-        "deterministic_server": True
+        "deterministic_server": True,
     }
     fw_utils.validate_mp_config(mp_parameters)