[BE] enable UFMT for torch/nn/*.py (#128593)

Part of #123062

- #123062
Pull Request resolved: https://github.com/pytorch/pytorch/pull/128593
Approved by: https://github.com/mikaylagawarecki

torch/_utils_internal.py

@@ -9,6 +9,7 @@ from typing import Any, Dict, Optional
 import torch
 from torch._strobelight.compile_time_profiler import StrobelightCompileTimeProfiler
 
+
 log = logging.getLogger(__name__)
 
 if os.environ.get("TORCH_COMPILE_STROBELIGHT", False):
@@ -133,7 +134,10 @@ def check_if_torch_exportable():
 
 
 def log_torch_jit_trace_exportability(
-    api: str, type_of_export: str, export_outcome: str, result: str
+    api: str,
+    type_of_export: str,
+    export_outcome: str,
+    result: str,
 ):
     _, _, _, _ = api, type_of_export, export_outcome, result
     return