fix: fix formatting

Author: viclzhu

src/sagemaker/estimator.py

@@ -3851,7 +3851,9 @@ def _distribution_configuration(self, distribution):
             # first make sure torch_distributed is enabled if instance type is p5
             torch_distributed_enabled = False
             if "torch_distributed" in distribution:
-                torch_distributed_enabled = distribution.get("torch_distributed").get("enabled", False)
+                torch_distributed_enabled = distribution.get("torch_distributed").get(
+                    "enabled", False
+                )
             smdistributed = distribution["smdistributed"]
             smdataparallel_enabled = smdistributed.get("dataparallel", {}).get("enabled", False)
             p5_enabled = "p5.48xlarge" in self.instance_type