format doc

Author: YY4994

src/flag_gems/runtime/precision_register.py

@@ -162,18 +162,12 @@ def wrapper(*args, **kwargs):
             if not is_close:
                 cfg["logged_ops"].add(op_key)
                 logger = get_precision_logger()
-                input_info = [
-                    get_tensor_info(a) for a in args if get_tensor_info(a)
-                ]
+                input_info = [get_tensor_info(a) for a in args if get_tensor_info(a)]
                 output_info = get_tensor_info(fg_result)
 
-                msg = (
-                    f"Op: {op_key} | FAIL | in: {input_info} | out: {output_info}"
-                )
+                msg = f"Op: {op_key} | FAIL | in: {input_info} | out: {output_info}"
                 if "error" in info:
-                    msg += (
-                        f" | {info['error']}: fg={info['fg']}, pt={info['pt']}"
-                    )
+                    msg += f" | {info['error']}: fg={info['fg']}, pt={info['pt']}"
                 else:
                     msg += (
                         f" | max_abs: {info['max_abs']:.6e}"
@@ -207,4 +201,4 @@ def register_impl(self, key, fn):
         device_key = self.reg_key
         self.all_ops.append(fn.__name__)
         self.all_keys.append(key)
-        self.lib.impl(key, wrapped_fn, device_key)
+        self.lib.impl(key, wrapped_fn, device_key)

src/flag_gems/runtime/register.py

@@ -142,4 +142,4 @@ def __getattr__(name):
         from .precision_register import PrecisionCheckRegister
 
         return PrecisionCheckRegister
-    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")