[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

transformer_engine/pytorch/fp8.py

@@ -485,9 +485,9 @@ def restore_fp8_meta_tensors(fp8_meta: Dict[str, Any]) -> None:
 
 @contextmanager
 def fp8_model_init(
-        enabled: bool = True,
-        preserve_high_precision_init_val: bool = False,
-    ) -> None:
+    enabled: bool = True,
+    preserve_high_precision_init_val: bool = False,
+) -> None:
     """
     Context manager for FP8 initialization of parameters.
 

transformer_engine/pytorch/module/base.py

@@ -882,6 +882,7 @@ def reset_parameters(self, defer_init: Optional[bool] = False) -> None:
             #       a parameter so we always re-apply it just for extra safety.
             param = torch.nn.Parameter(param)
             if self.primary_weights_in_fp8 and self.preserve_high_precision_init_val:
+
                 def get(self):
                     if hasattr(self, "_high_precision_init_val"):
                         return self._high_precision_init_val